Skip to content
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

Support New Methods and Fix Bugs (V0.2.4) #7

Closed
Lupin1998 opened this issue Jul 2, 2022 · 1 comment
Closed

Support New Methods and Fix Bugs (V0.2.4) #7

Lupin1998 opened this issue Jul 2, 2022 · 1 comment
Assignees
Labels
bug Something isn't working enhancement New feature or request update Updating configs and results in OpenMixup

Comments

@Lupin1998
Copy link
Member

Lupin1998 commented Jul 2, 2022

Supporting new methods:

  1. Support new backbone architectures (LITv2).
  2. Refactor code structures weight initialization in various network modules (using BaseModule in mmcv).
  3. Refactor code structures of openmixup.models.utils.layers to support more network structures.

Fixing bugs:

  1. Fix bugs that cause degenerate performances of pure Transformer backbones (DeiT and Swin) in OpenMixup. The main reason might be the old version of auto_fp16 and DistOptimizerHook implementations, since PyTorch=>1.6.0 has better support of fp16 training than mmcv.
  2. Fix the bug of ViT fine-tuning for MIM methods (e.g., MAE, SimMIM). The original MIMVisionTransformer in openmixup.models.mim_vit has frozen all the backbone parameters during fine-tuning.
  3. Fix the weight initialization of Transformer-based architectures (e.g., ViT, Swin) to reproduce the train-from-scratch performance. Update weight initialization and fp16 training settings in relevant config files.
@Lupin1998 Lupin1998 added bug Something isn't working enhancement New feature or request update Updating configs and results in OpenMixup labels Jul 2, 2022
@Lupin1998 Lupin1998 self-assigned this Jul 2, 2022
@Lupin1998 Lupin1998 pinned this issue Jul 2, 2022
@Lupin1998
Copy link
Member Author

Lupin1998 commented Jul 7, 2022

OpenMixup v0.2.4 is now released, which supports various Transformer architectures and fixes bugs.

@Lupin1998 Lupin1998 unpinned this issue Aug 14, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working enhancement New feature or request update Updating configs and results in OpenMixup
Projects
None yet
Development

No branches or pull requests

1 participant