From 4c3b92478eb1ac23636f6977d5bdc633b19f8f3f Mon Sep 17 00:00:00 2001 From: Bhaskar Gautam Date: Sun, 4 Nov 2018 22:46:21 +0530 Subject: [PATCH] Resolved multiple imbalance GPU devices issue. --- train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train.py b/train.py index f9285a7..d1bfd4c 100755 --- a/train.py +++ b/train.py @@ -57,6 +57,8 @@ experiment.add_argument('--val_interval', type=int, default=200, help='how many steps to wait before vaidation [default: 200]') experiment.add_argument('--save_interval', type=int, default=1, help='how many epochs to wait before saving [default:1]') +import os +os.environ['CUDA_VISIBLE_DEVICES']='0' # Add your Device ID in case of multiple GPU devices (imbalance) def train(train_loader, dev_loader, model, args):