-
Notifications
You must be signed in to change notification settings - Fork 27.3k
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
remove the obsolete code related to fairscale FSDP #26651
Conversation
@muellerzr and @ArthurZucker could you please take a look at this PR? Thanks :D |
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! Looks good to me
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
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.
Sounds good, let's rather raise an error here saying that it is no longer supported!
" `--half_precision_backend apex`: GPU bf16 is not supported by apex. Use" | ||
" `--half_precision_backend cuda_amp` instead" | ||
) | ||
raise ValueError(" `--half_precision_backend apex`: GPU bf16 is not supported by apex.") |
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 error info. cc @ArthurZucker
@amyeroberts sorry for bothering, could you please review this PR? |
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 updating the code and iterating!
* remove the obsolete code related to fairscale FSDP * apple review suggestion
What does this PR do?
As the title says, this PR introduces two modifications:
--bf16
and--half_precision_backend apex
as thecuda_amp
option is no longer available. I'm not sure how to write it more appropriately, so I just useauto
instead.