Merge pull request #2 from SsnL/comma

Add missing comma in argparse option
znxlwm · Nov 21, 2017 · 5207b2f · 5207b2f
2 parents 9ffe72e + 7c0f1f7
commit 5207b2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -16,7 +16,7 @@ def parse_args():
     parser = argparse.ArgumentParser(description=desc)
 
     parser.add_argument('--gan_type', type=str, default='EBGAN',
-                        choices=['GAN', 'CGAN', 'infoGAN', 'ACGAN', 'EBGAN', 'BEGAN', 'WGAN', 'WGAN_GP' 'DRAGAN', 'LSGAN'],
+                        choices=['GAN', 'CGAN', 'infoGAN', 'ACGAN', 'EBGAN', 'BEGAN', 'WGAN', 'WGAN_GP', 'DRAGAN', 'LSGAN'],
                         help='The type of GAN')#, required=True)
     parser.add_argument('--dataset', type=str, default='mnist', choices=['mnist', 'fashion-mnist', 'celebA'],
                         help='The name of dataset')