updated args

2026-09-09 11:32:07 +08:00 · 2019-06-25 18:51:41 -04:00
parent de0f7fc936
commit d4ca295762
2 changed files with 7 additions and 4 deletions
@@ -98,6 +98,7 @@ def main(hparams, cluster, results_dict):
        cluster=cluster,
        checkpoint_callback=checkpoint,
        early_stop_callback=early_stop,
+        gpus=hparams.gpus
    )

    # train model