Skip to content

Instantly share code, notes, and snippets.

@williamFalcon
Created July 29, 2019 17:53
Show Gist options
  • Save williamFalcon/252ac70870ba54b6790a47a78f8ea049 to your computer and use it in GitHub Desktop.
Save williamFalcon/252ac70870ba54b6790a47a78f8ea049 to your computer and use it in GitHub Desktop.
# train on 4 gpus on the same machine MUCH faster than DataParallel
trainer = Trainer(distributed_backend='ddp', gpus=[0, 1, 2, 3])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment