added single node distdataparallel

2026-09-26 14:11:01 +08:00 · 2019-07-03 16:34:49 -04:00
parent 7010d16752
commit 32eddf492e
2 changed files with 4 additions and 1 deletions
@@ -101,6 +101,7 @@ def main(hparams, cluster, results_dict):
        checkpoint_callback=checkpoint,
        early_stop_callback=early_stop,
        gpus=gpu_list,
+        nb_gpu_nodes=1
    )

    # train model