From 41f68861d505f224cb3066b9a6239d538c4d764b Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 25 Jun 2019 18:42:44 -0400
Subject: [PATCH] updated args

---
 docs/source/examples/fully_featured_trainer.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/docs/source/examples/fully_featured_trainer.py b/docs/source/examples/fully_featured_trainer.py
index 329fe5ba00..736fa28422 100644
--- a/docs/source/examples/fully_featured_trainer.py
+++ b/docs/source/examples/fully_featured_trainer.py
@@ -188,10 +188,10 @@ if __name__ == '__main__':
 
     # single or multiple GPUs on same machine
     gpu_ids = hyperparams.gpus.split(';')
-    if len(gpu_ids) == 1:
+    if hyperparams.live:
         # run on 1 gpu
-        print(f'RUNNING 1 TRIAL ON GPU. gpu: {gpu_ids[0]}')
-        os.environ["CUDA_VISIBLE_DEVICES"] = gpu_ids[0]
+        print(f'RUNNING INTERACTIVE TRIAL ON GPUS. gpu ids: {gpu_ids}')
+        os.environ["CUDA_VISIBLE_DEVICES"] = gpu_ids
         main(hyperparams, None, None)
 
     else: