Alpha-VL / ConvMAE

ConvMAE: Masked Convolution Meets Masked Autoencoders
MIT License
484 stars 41 forks source link

Model Settings and checkpoint not match #32

Open z-jiaming opened 1 year ago

z-jiaming commented 1 year ago

Thanks for your great work!

But I have a problem about the model setting with your provided checkpoint. I load your checkpoints, but the model setting that can be loaded correctly does not match what is written in the paper.

  1. the mlp_ratio of Large and Huge
  2. the patch_size of huge

I want to find out what's going on, thanks a lot!

image image