search

Haoqing-Wang / LocalMIM

[CVPR 2023 Highlight] Masked Image Modeling with Local Multi-Scale Reconstruction
40 stars 3 forks source link

A typo? #4

Closed utkutpcgl closed 9 months ago

utkutpcgl commented 9 months ago

I guess there is a minor typo in the pretrain code: https://github.com/Haoqing-Wang/LocalMIM/blob/314a6e558a8c47bf0ef146235e2ced8d01e27fe6/ViT/run_pretrain.py#L44

This should be rather "MIM_vit_large_patch16" I guess?

Haoqing-Wang commented 9 months ago

Yes, thanks for reminding me. In fact, this argument is typically defined in the running script.