Update train_txt2im.py

zsdonghao · Feb 9, 2018 · bdba1de · bdba1de
1 parent ac7b989
commit bdba1de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_txt2im.py b/train_txt2im.py
@@ -156,7 +156,7 @@ def main_train():
         # print(sample_sentence[i])
     sample_sentence = tl.prepro.pad_sequences(sample_sentence, padding='post')
 
-    n_epoch = 600
+    n_epoch = 100 # 600
     print_freq = 1
     n_batch_epoch = int(n_images_train / batch_size)
     # exit()