diff --git a/examples/multi_node_examples/multi_node_demo.py b/examples/multi_node_examples/multi_node_demo.py index 577336b8..c380497b 100644 --- a/examples/multi_node_examples/multi_node_demo.py +++ b/examples/multi_node_examples/multi_node_demo.py @@ -30,7 +30,8 @@ def main(hparams): # ------------------------ trainer = Trainer( gpus=2, - nb_gpu_nodes=2 + nb_gpu_nodes=2, + distributed_backend='ddp' ) # ------------------------