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

Porting model weight #131

Open
asd841018 opened this issue Mar 29, 2024 · 1 comment
Open

Porting model weight #131

asd841018 opened this issue Mar 29, 2024 · 1 comment

Comments

@asd841018
Copy link

Hi, thank you for the awesome work.

I notice that in your Ported Models (e.g. [r100 12m AdaFace]) description, you ported the original repo's weight to tensorflow version.

May I ask how did you do that? Is it possible to port different deep learning framework weight?

@leondgarse
Copy link
Owner

  • I'm keeping the porting script here 08-20_pytorch_to_keras.md#adaface-model.
  • Ya, porting weight is not that hard. Technically, it's just loading pytorch weights as list -> align weights order matching keras model -> loading weights into keras model. You can also see in my another repo keras_cv_attention_models, that most models there are ported from pytorch, also some ported from paddlepaddle and mindspore. Most porting script also kept in my notebook above.

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