不匹配
ats4869 opened this issue · comments
Yufei Zhang commented
pretrain使用的是maemodel模型,finetune使用的是vitmodel模型,我的理解对吧,如果是这样的话pretrain的模型怎么能够再次finetune呢?
dreamcubeblock commented
mae的encoder就是vit啊,把vit的参数加载了就能finetune了,好好去看看代码
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
ats4869 opened this issue · comments
pretrain使用的是maemodel模型,finetune使用的是vitmodel模型,我的理解对吧,如果是这样的话pretrain的模型怎么能够再次finetune呢?
mae的encoder就是vit啊,把vit的参数加载了就能finetune了,好好去看看代码