-
Notifications
You must be signed in to change notification settings - Fork 1.6k
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
[Feature Request] Apex support for mixed precision training #1719
Comments
nijkah
changed the title
Apex support for mixed precision training
[Feature Request] Apex support for mixed precision training
Feb 13, 2022
Thanks for your advice, but what is the advantage of apex over torch.cuda.amp? mmcv implements mixed precision training via Fp16OptimizerHook. |
@HAOCHENYE |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Describe the feature
Motivation
Many recent research works are based on apex for mixed precision training.
So I want to add this feature to mmcv with the name of class EpochBasedRunnerApex.
I am thinking implementation of just wrapping EpochBasedRunner as in Swin-Transformer Implementation.
Is it good idea to mmcv? Then I will create the PR.
Related resources
https://github.com/SwinTransformer/Swin-Transformer-Object-Detection
https://github.com/facebookresearch/ConvNeXt
https://github.com/microsoft/unilm/tree/master/beit/semantic_segmentation
The text was updated successfully, but these errors were encountered: