-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Added eps attribute to FrozenBatchNorm2d #2190
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
Conversation
|
@fmassa I'm not sure where the error comes from, it seems like the tests timed out in |
|
Also, I made two choices here that any reviewer might want to take a look at:
|
For the first option, I'm not sure if it introduces a BC breaking because it just re-defines the name of the argument. |
fmassa
left a comment
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.
Thanks for the PR!
I have a couple of comments, could you look into addressing those?
|
Test failures seem related |
|
Thanks, I fixed those. There is a timeout remaining for windows builds and an unrelated (I think) failure on travis! |
fmassa
left a comment
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.
Thanks a lot!
* feat: Added eps argument to FrozenBatchNorm2d * test: Added unittest for eps addition in FrozenBatchNorm2d See pytorch#2169 * fix: Reverted forward changes for JIT fuser * fix: Added back n argument for backward-compatibility * fix: Fixed FrozenBatchNorm2d forward Added back eps * feat: Specified deprecation warnings in FrozenBatchNorm2d * test: Added unittest for deprecation warninig in FrozenBatchNorm2d * style: Fixed lint * style: Fixed block comment lint
This PR aims at tackling #2169 by:
epsattribute to ̀FrozenBatchNorm2d`npositional argument ofFrozenBatchNorm2dconstructor tonum_featuresBatchNorm2dforward in eval mode.