Update train_cycleGAN_loss.py

Tandon-A · Nov 22, 2018 · cca9dd8 · cca9dd8
1 parent 007bd3c
commit cca9dd8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_cycleGAN_loss.py b/train_cycleGAN_loss.py
@@ -150,7 +150,7 @@ def main(_):
         lr_rate = 0.0002
         beta1 = 0.5
         max_img = 100    
-        # change loss type. Options - l1, l2, ssim, ssim_1l, ssim_l2_a, ssim_l2_b, ssim_l1l2_a, ssim_l1l2_b, l1_l2
+        # change loss type. Options - l1, l2, ssim, ssim_l1, ssim_l2_a, ssim_l2_b, ssim_l1l2_a, ssim_l1l2_b, l1_l2
         loss_type = FLAGS.loss_type
         tf.reset_default_graph()