delete only_d

yifanjiang19 · Dec 15, 2018 · ccb413d · ccb413d
1 parent da6f55b
commit ccb413d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/train.py b/train.py
@@ -16,7 +16,7 @@
 CRITIC_ITERS = 5
 total_steps = 0
 iter_d = 0
-only_d = True
+only_d = False
 
 for epoch in range(1, opt.niter + opt.niter_decay + 1):
     epoch_start_time = time.time()
@@ -28,10 +28,10 @@
         # if w[0]-x[0] < 39:
         #     continue
         model.set_input(data)
-        if iter_d <= CRITIC_ITERS-1:
-            only_d = False
-        else:
-            only_d = False
+#         if iter_d <= CRITIC_ITERS-1:
+#             only_d = False
+#         else:
+#             only_d = False
         model.optimize_parameters(only_d)
 
         if total_steps % opt.display_freq == 0: