Merge pull request #69 from kopytjuk/bugfix-windows

Bugfix to run the code on windows machines.
2018-12-25 11:43:31 -05:00 · 2018-12-25 11:43:31 -05:00 · 72853533a0
parent f81eafd2aa afd82ac354
commit 72853533a0
1 changed files with 192 additions and 190 deletions
--- a/main.py
+++ b/main.py
@ -16,6 +16,8 @@ import os
 import models.dcgan as dcgan
 import models.mlp as mlp

+if __name__=="__main__":
+
    parser = argparse.ArgumentParser()
    parser.add_argument('--dataset', required=True, help='cifar10 | lsun | imagenet | folder | lfw ')
    parser.add_argument('--dataroot', required=True, help='path to dataset')