hello, a little question about this code.

[thebestYezhang]

I am beginner and crazy about this MAE, but about the pretrained model--pretrain_mae_vit_base_mask_0.75_400e.pth downloaded from BaiDu yun, I really wanna know where to load it in the file. Can it be used in modeling_pretrain.py?

[pengzhiliang]

Yes, maybe the visualization is a good start.

[thebestYezhang]

不好意思，我没懂您的意思。是这样的，我想加载你们的MAE训练好的模型，也就是pretrain_mae_vit_base_mask_0.75_400e.pth这个权重。然后取出encoder部分进行finetune，然后去做我的分类任务，这样子应该怎么操作呢？指导一下，感激不尽！！！

[pengzhiliang]

Ok. You can fine tune it on your dataset.
Please reference: run_class_finetuning.py

[zsddd]

不好意思，我没懂您的意思。是这样的，我想加载你们的MAE训练好的模型，也就是pretrain_mae_vit_base_mask_0.75_400e.pth这个权重。然后取出encoder部分进行finetune，然后去做我的分类任务，这样子应该怎么操作呢？指导一下，感激不尽!!!

你好，请问你这个分类任务成功的吗？可否提供下代码，感谢