search

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification
https://classyvision.ai
MIT License
1.59k stars 278 forks source link

Pytorch AMP #666

Closed blefaudeux closed 3 years ago

blefaudeux commented 3 years ago

Summary: Add Pytorch AMP support. A follow up for FairScale would be to add ShardedGradScaler, so that we support mixed precision with ShardedDDP and ShardedOptimizer

Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D25383305