-
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
Add Metric <-> Lightning Module integration tests #4008
Conversation
Hello @teddykoker! Thanks for updating this PR. There are currently no PEP 8 issues detected in this Pull Request. Cheers! 🍻 Comment last updated at 2020-10-09 01:27:55 UTC |
Codecov Report
@@ Coverage Diff @@
## master #4008 +/- ##
=======================================
+ Coverage 87% 89% +2%
=======================================
Files 212 212
Lines 15499 15562 +63
=======================================
+ Hits 13519 13873 +354
+ Misses 1980 1689 -291 |
return self.step(x) | ||
|
||
def training_epoch_end(self, outs): | ||
assert torch.allclose(self.sum, self.metric.compute()) |
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.
now much close, lets spec tolerance, also why not exact?
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.
default tolerance is 1e-8, i can parameterize if you want. Can't do exact bc of floating point
e971945
to
73338d0
Compare
73338d0
to
e517ac9
Compare
Test Lightning integration with metrics