facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

不匹配

ats4869 opened this issue · comments

pretrain使用的是maemodel模型,finetune使用的是vitmodel模型,我的理解对吧,如果是这样的话pretrain的模型怎么能够再次finetune呢?

mae的encoder就是vit啊,把vit的参数加载了就能finetune了,好好去看看代码