The parameters of the model cannot be matched

bair-climate-initiative / scale-mae

Make your models invariant to changes in scale.

https://ai-climate.berkeley.edu/scale-mae-website/

Other

121 stars 13 forks source link

The parameters of the model cannot be matched #17

Open Weeb-killer opened 1 month ago

Weeb-killer commented 1 month ago

Thanks for your great work! When I use main_linearprobe.py, I find that "mae_vit_large_patch16" does not exist in models_vit.py. Do I need to change this to "vit_large_patch16"?