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

About loading the ImageNet Pretrain #7

Open
fuyimin96 opened this issue Nov 4, 2022 · 2 comments
Open

About loading the ImageNet Pretrain #7

fuyimin96 opened this issue Nov 4, 2022 · 2 comments

Comments

@fuyimin96
Copy link

  1. Hi, when I saw your code, I found that only the first 11 layers of the transformer are loaded (when feature_fusion == True)
    And the "ff_last_layer" and "ff_encoder_norm" are trained from scratch, am I right?
  2. If so, what is the performance when loading 12th layer weights to ff_last_layer and norm to off_encoder_norm?
    Thanks
@Markin-Wang
Copy link
Owner

Hi, thanks for your interest.
Yes exactly, we only load the pretrained weights of the first 11 layers when feature fusion is enabled. The intuition behind is that the input distribution of the ff_last_layer and the pretraiend layer is quite different.
Sorry, we didn't conduct the experients of loading all the 12 layers weights. Loading the pretrained norm weights give slightly worser results on some datasets if I remember correctly.

@fuyimin96
Copy link
Author

Thanks for your reply

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