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

[Feature]training timm model with frozen_stages and norm eval tricks #605

Closed
Bo396543018 opened this issue Dec 15, 2021 · 8 comments
Closed

Comments

@Bo396543018
Copy link

I found use frozen_stages&norm eval may improve transfer learning performance.
Currently,the timm wrapper not support these tricks.

@Ezra-Yu
Copy link
Collaborator

Ezra-Yu commented Dec 15, 2021

Yes, they do not implement related functions. Every backbone has a different implement. Currently, there is no way to modify this in general.

what backbone do you want to use?

@Bo396543018
Copy link
Author

I use timm's resnet50 model to compare 21k &1k pretrained improvement.

@mzr1996
Copy link
Member

mzr1996 commented Dec 15, 2021

So why don't just use mmcls ResNet implementation

@Bo396543018
Copy link
Author

Use timm warpper could avoid the problem for 21k pretrained weights inconsistent naming.

@mzr1996
Copy link
Member

mzr1996 commented Dec 15, 2021

Well, the freeze stages function depends on the detailed network and we cannot provide a unified API except timm provides it.

@Bo396543018
Copy link
Author

Ok,Thank you for your reply,I'll try to convert weights for comparison.

@Ezra-Yu
Copy link
Collaborator

Ezra-Yu commented Dec 15, 2021

@Bo396543018 We are considering adding relevant pre-trained models to mmcls, and we will thank you if you complete the relevant work, put relevant weights or scripts in mmcls.

@Bo396543018
Copy link
Author

@ Ezra-Yu OK,When I complete,I'll share the converted weights

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants