Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adds back the slow spawn ddp implementation that people want #2115

Merged
merged 28 commits into from
Jun 8, 2020

Conversation

williamFalcon
Copy link
Contributor

@williamFalcon williamFalcon commented Jun 8, 2020

the implementation with .spawn has many issues

  • original model does not retain weights.
  • dataloader num_workers doesn't scale well.

This PR adds back this implementation under 'ddp_spawn' but keeps the much faster, script-based version 'ddp' as default.

FIxes #2116
Fixes #2104 (use ddp_spawn in the meantime)
Fixes #2092
Fixes #2086
Fixes #2085

@pep8speaks
Copy link

pep8speaks commented Jun 8, 2020

Hello @williamFalcon! Thanks for updating this PR.

There are currently no PEP 8 issues detected in this Pull Request. Cheers! 🍻

Comment last updated at 2020-06-08 21:36:29 UTC

@mergify mergify bot requested a review from a team June 8, 2020 15:37
@codecov
Copy link

codecov bot commented Jun 8, 2020

Codecov Report

Merging #2115 into master will decrease coverage by 0%.
The diff coverage is 14%.

@@          Coverage Diff           @@
##           master   #2115   +/-   ##
======================================
- Coverage      86%     86%   -0%     
======================================
  Files          75      75           
  Lines        4786    4791    +5     
======================================
  Hits         4129    4129           
- Misses        657     662    +5     

@Borda Borda added bug Something isn't working feature Is an improvement or enhancement labels Jun 8, 2020
@Borda Borda added this to the 0.8.0 milestone Jun 8, 2020
@williamFalcon williamFalcon merged commit 3260e59 into master Jun 8, 2020
@Borda Borda deleted the old_ddp branch June 8, 2020 22:18
justusschock pushed a commit that referenced this pull request Jun 29, 2020
* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* training batch clean up

* adding spawn

* adding spawn

* adding spawn

* adding spawn

* adding spawn

* adding spawn

* adding spawn

* adding spawn
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment