minor update

txWang · Jul 17, 2019 · 859516a · 859516a
1 parent 5e41bac
commit 859516a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/main_pancreas.py b/main_pancreas.py
@@ -24,7 +24,7 @@
 # nn parameter
 code_dim = 20
 batch_size = 50 # batch size for each cluster
-num_epochs = 200
+num_epochs = 2000
 base_lr = 1e-3
 lr_step = 200  # step decay of learning rates
 momentum = 0.9

diff --git a/main_pbmc.py b/main_pbmc.py
@@ -25,7 +25,7 @@
 # nn parameter
 code_dim = 20
 batch_size = 50 # batch size for each cluster
-num_epochs = 200
+num_epochs = 2000
 base_lr = 1e-3
 lr_step = 200  # step decay of learning rates
 momentum = 0.9