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

Pretrained weights for other versions #46

Closed
JRestom opened this issue Jan 12, 2023 · 8 comments
Closed

Pretrained weights for other versions #46

JRestom opened this issue Jan 12, 2023 · 8 comments

Comments

@JRestom
Copy link

JRestom commented Jan 12, 2023

Are there any available weights for the PoolFormer version with BN and LN instead of the Modified Layer Norm? (Table 5 in https://arxiv.org/abs/2111.11418)

@yuweihao
Copy link
Collaborator

Hi @JRestom ,

Thanks for your attention. For BN checkpoint, I have posted it in this issue #5 . For LN checkpoint, sorry that it is deleted. I will retrain it in a week and share it with you. Thanks.

@JRestom
Copy link
Author

JRestom commented Jan 13, 2023

Thank you! It would be of great help.

@yuweihao
Copy link
Collaborator

yuweihao commented Jan 22, 2023

Hi @JRestom ,

poolformer_ln_s12.pth.tar (let norm_layer=LayerNormChannel) is released.

Thanks!

@JRestom
Copy link
Author

JRestom commented Jan 22, 2023

Hello @yuweihao,
I highly appreciate the weights. One issue, when opening the link it directs me to a file called poolformer_bn_s12 (BatchNorm I assume). Could you double check the link?

@yuweihao
Copy link
Collaborator

Hi @JRestom , yes, poolformer_bn_s12.pth.tar and let norm_layer=torch.nn.BatchNorm2d.

@JRestom
Copy link
Author

JRestom commented Jan 22, 2023

I meant, the link of poolformer_ln_s12.pth.tar is directing me to the same folder as poolformer_bn_s12.pth.tar. Shouldn't they be different? (one for LayerNom and one for BatchNorm).

@yuweihao
Copy link
Collaborator

Oh, my fault, the right link is poolformer_ln_s12.pth.tar.

@JRestom
Copy link
Author

JRestom commented Jan 22, 2023

Thank you so much!!

@yuweihao yuweihao closed this as completed Feb 9, 2023
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