updated args

2019-06-25 18:58:45 -04:00 · 2019-06-25 18:58:45 -04:00 · bf3b86ce4d
parent 684dfd0a38
commit bf3b86ce4d
1 changed files with 1 additions and 1 deletions
--- a/docs/source/examples/fully_featured_trainer.py
+++ b/docs/source/examples/fully_featured_trainer.py
@ -95,7 +95,7 @@ def main(hparams, cluster, results_dict):
    # gpus are ; separated for inside a node and , within nodes
    gpu_list = None
    if hparams.gpus is not None:
-        gpu_list = map(int, hparams.gpus.split(';'))
+        gpu_list = [int(x) for x in hparams.gpus.split(';')]

    # configure trainer
    trainer = Trainer(