Update params.py

mashaan14 · Jul 10, 2023 · 0d97c4d · 0d97c4d
1 parent 0e99993
commit 0d97c4d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/params.py b/params.py
@@ -2,17 +2,15 @@
 
 # params for training network
 num_gpu = 1
-num_epochs_pre = 1000
+num_epochs_pre = 1000 #10
 save_step_pre = 100
-num_epochs = 1000
+num_epochs = 2000 #100
 log_step = 100
 save_step = 100
 manual_seed = None
 
 
 # params for optimizing models
 batch_size = 32
-d_lr = 1e-4 #1e-4
-g_lr = 1e-4 #1e-4
-beta1 = 0.5
-beta2 = 0.9
+d_lr = 2*1e-4 #1e-4
+g_lr = 2*1e-4 #1e-4