transformers
085bf5c1
- [trainer] add `--optim adamw_torch_fused` for pt-2.0+ (#22144)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[trainer] add `--optim adamw_torch_fused` for pt-2.0+ (#22144) * [trainer] add --optim adamw_torch_fused * change optim default * deal with non-torch * revert default change; prep; add fp16/amp assert * typo * typo
References
#22144 - [trainer] add `--optim adamw_torch_fused` for pt-2.0+
Author
stas00
Parents
c6318c37
Loading