-
Notifications
You must be signed in to change notification settings - Fork 3.4k
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
updated sync bn #2838
updated sync bn #2838
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Update trainer args so drone passes
9dadb29
to
66f948b
Compare
@williamFalcon not sure why the drone test is failing, the same test passes locally and was passing in the previous commit |
we don’t need that anymore since u added the test using ddp spawn no? |
ddp_spawn test is called using the example, as a script test |
the test is the example, maybe we shall put it to test as some other people crossing over would raise the same question... |
Hello @ananyahjha93! Thanks for updating this PR. There are currently no PEP 8 issues detected in this Pull Request. Cheers! 🍻 Comment last updated at 2020-08-05 22:55:59 UTC |
Codecov Report
@@ Coverage Diff @@
## master #2838 +/- ##
=======================================
+ Coverage 86% 90% +4%
=======================================
Files 78 78
Lines 7029 7029
=======================================
+ Hits 6049 6338 +289
+ Misses 980 691 -289 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nice addition @ananyahjha93 !
|
||
def prepare_data(self): | ||
# download only | ||
MNIST(self.data_dir, train=True, download=True, normalize=(0.1307, 0.3081)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just to double check, do we need TrialMNIST or MNIST?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I did not want to change TrialMNIST structure for this PR as I had to include a DistributedSampler with shuffle
dm.prepare_data() | ||
dm.setup(stage=None) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I thought the Trainer / LightningModule will call these automatically? @nateraw
What does this PR do?
Fixes # (issue)
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.
Did you have fun?
Make sure you had fun coding 🙃