make train_gan.py not totally broken by default

william-gx · May 3, 2017 · 46237d4 · 46237d4
1 parent 8301961
commit 46237d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_gan.py b/train_gan.py
@@ -142,7 +142,7 @@
     help='Dataset (mnist/imagenet)')
 parser.add_argument('--raw_size', type=int,
     help='Raw minor edge size (for image datasets)')
-parser.add_argument('--crop_size', type=int,
+parser.add_argument('--crop_size', type=int, default=28,
     help='Crop size (for image datasets)')
 parser.add_argument('--crop_resize', type=int,
     help='Size of crop for D in, G out (E still takes crop_size resolution)')