updated args

2026-09-09 11:32:07 +08:00 · 2019-06-25 18:58:45 -04:00
parent 684dfd0a38
commit bf3b86ce4d
1 changed files with 1 additions and 1 deletions
@@ -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(