Fixed nll with label_smoothing to just nll#28708
Conversation
younesbelkada
left a comment
There was a problem hiding this comment.
Thanks ! Could you apply the same changes on the TF modeling file?
b22eaf9 to
5eb9e3c
Compare
|
@younesbelkada , |
younesbelkada
left a comment
There was a problem hiding this comment.
Thanks ! there is a small merge conflict with main - could you resolve that ? After that well be able to merge
5eb9e3c to
90d2459
Compare
|
@younesbelkada , |
younesbelkada
left a comment
There was a problem hiding this comment.
Thank you @nileshkokane01 for your contribution !
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
ArthurZucker
left a comment
There was a problem hiding this comment.
IMO that's a breaking change. Not sure if the best is to update the documentation or update this!
Should we add a config arg for this?
|
@ArthurZucker IMO this is not really breaking, it is even the opposite as it fixes some subtle bugs with respect to training with BLIP- see @NielsRogge ' comment here: #28167 (comment) |
|
thinking a bit about it, indeed maybe we should make that configurable through a variable in the config so that potentially users could revert to original behaviour if needed. |
|
@younesbelkada sure! I'll do that. |
|
thank you @nileshkokane01 ! |
|
Do I have to change the nll loss to the following as well:
|
|
hi @nileshkokane01 |
90d2459 to
03f72e6
Compare
|
@younesbelkada , |
younesbelkada
left a comment
There was a problem hiding this comment.
Looking good thanks ! left few nits on docstrings!
amyeroberts
left a comment
There was a problem hiding this comment.
Thanks for fixing and making this controllable!
What does this PR do?
This PR fixes #28167 by making label_smoothing= 0 .
Fixes #28167
Before submitting
Pull Request section?
to it if that's the case.
documentation guidelines, and
here are tips on formatting docstrings.
Who can review?
@younesbelkada
Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.