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

Pre-trained model loading error #10

Closed
bobo0810 opened this issue Jun 23, 2021 · 2 comments
Closed

Pre-trained model loading error #10

bobo0810 opened this issue Jun 23, 2021 · 2 comments

Comments

@bobo0810
Copy link

Thank you very much for this work!

Inconsistent quantities caused the loading to fail. Do you know why? Thanks~

  • tasks/distfc/README.md IR101 pretrained model: OrderedDict len is 672
  • torchkit/backbone/model_irse.py IR101:OrderedDict len is 917
@HuangYG123
Copy link
Collaborator

Sorry for the inconsistency. This pre-trained model is trained using the code of the original paper version. You can use load this pre-train model with the model definition file in https://github.com/HuangYG123/CurricularFace/blob/master/backbone/model_irse.py.
We will release new pre-trained models with this version of the code soon.

@bobo0810
Copy link
Author

Sorry for the inconsistency. This pre-trained model is trained using the code of the original paper version. You can use load this pre-train model with the model definition file in https://github.com/HuangYG123/CurricularFace/blob/master/backbone/model_irse.py.
We will release new pre-trained models with this version of the code soon.

Thanks:blush:

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

2 participants