cleaning up demos

This commit is contained in:
William Falcon
2019-10-05 15:05:23 -04:00
parent e2bf0cda07
commit f2fb219bfe
@@ -30,7 +30,8 @@ def main(hparams):
# ------------------------
trainer = Trainer(
gpus=2,
nb_gpu_nodes=2
nb_gpu_nodes=2,
distributed_backend='ddp'
)
# ------------------------