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