From 0144f13e55209e9f31294812062568d0337dab95 Mon Sep 17 00:00:00 2001 From: jacqueline Date: Tue, 12 Mar 2019 10:40:36 +0800 Subject: [PATCH] Update test_GFN_4x.py --- test_GFN_4x.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test_GFN_4x.py b/test_GFN_4x.py index 4776796..3e5c291 100755 --- 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=[