Merge pull request #1 from shalgi/patch-2

Update config.py
shalgi · Jan 8, 2020 · 4452869 · 4452869
2 parents 4686249 + 4531e29
commit 4452869
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/config.py b/config.py
@@ -6,6 +6,7 @@ def get_arguments():
     #parser.add_argument('--mode', help='task to be done', default='train')
     #workspace:
     parser.add_argument('--not_cuda', action='store_true', help='disables cuda', default=0)
+    parser.add_argument('--cuda_id', help='the cuda id of the GPU to train on',default='0')
 
     #load, input, save configurations:
     parser.add_argument('--netG', default='', help="path to netG (to continue training)")