-
Notifications
You must be signed in to change notification settings - Fork 195
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
Fix PolynomialLR power type. #1440
Fix PolynomialLR power type. #1440
Conversation
@dotnet-policy-service agree company="HACARUS" |
regarding |
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.
Hey, LGTM, thanks for the fix :). Can you update your branch and also add the some notes in Releasenotes ?
It is a flaky test, we will try running again, don't worry |
802c1ef
to
5b76bb7
Compare
|
5b76bb7
to
d1f9b3f
Compare
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.
Awesome, thanks 👍
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.
LGTM.
Thanks for the contribution.
Hey @hiyuh, one of your PRs got merged, thanks for the contribution! Can you fix the Releasenotes conflict so we can merge this one too ? |
torch.optim.lr_scheduler.PolynomialLR power was typed int, but should be double. Non-integer power is widely used for common training recipe. E.g. torchvision's pre-trained semantic segmentation models uses PolynomialLR as main LR scheduler with power = 0.9; https://github.com/pytorch/vision/blob/main/references/segmentation/train.py#L201 See also https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.PolynomialLR.html
d1f9b3f
to
79739db
Compare
@alinpahontu2912 |
torch.optim.lr_scheduler.PolynomialLR
power
was typedint
, but should bedouble
.Non-integer
power
is widely used for common training recipe.E.g.
torchvision
's pre-trained semantic segmentation models usesPolynomialLR
as main LR scheduler withpower = 0.9
; https://github.com/pytorch/vision/blob/main/references/segmentation/train.py#L201See also https://pytorch.org/docs/stable/generated/torch.optim.lr_scheduler.PolynomialLR.html