Update train_GFN_4x.py

jacquelinelala · Mar 12, 2019 · 3a8c7b1 · 3a8c7b1
1 parent 8c0655e
commit 3a8c7b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_GFN_4x.py b/train_GFN_4x.py
@@ -35,7 +35,7 @@
 parser.add_argument("--gated", type=bool, default=False, help="Activated gate module")
 parser.add_argument("--isTest", type=bool, default=False, help="Test or not")
 parser.add_argument('--dataset', required=True, help='Path of the training dataset(.h5)')
-os.environ['CUDA_VISIBLE_DEVICES'] = '1'
+os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 training_settings=[