Update test_GFN_4x.py

jacquelinelala · Mar 12, 2019 · 0144f13 · 0144f13
1 parent 3a8c7b1
commit 0144f13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_GFN_4x.py b/test_GFN_4x.py
@@ -29,7 +29,7 @@
 parser.add_argument("--isTest", type=bool, default=True, help="Test or not")
 parser.add_argument('--dataset', required=True, help='Path of the validation dataset')
 parser.add_argument("--intermediate_process", default="", type=str, help="Test on intermediate pkl (default: none)")
-os.environ['CUDA_VISIBLE_DEVICES'] = '1'
+os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 test_set=[