Recommendations for training poolFormer on custom dataset?

sail-sg / poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)

https://arxiv.org/abs/2111.11418

Apache License 2.0

1.3k stars 117 forks source link

Open loucif01 opened 2 years ago

loucif01 commented 2 years ago

Hello, Thank you for this great contribution. Can you please clarify recommendations for training on a custom dataset?

yuweihao commented 2 years ago

Hi @loucif01 ,

Sorry that I do not have much experience in training on other custom datasets. Maybe you can tune the learning rate and drop-path rate.