search

JeremyXSC / DMRL

MIT License
1 stars 0 forks source link

预训练文件问题 #2

Open SweetLittleLies opened 2 weeks ago

SweetLittleLies commented 2 weeks ago

您好，我按照您的提示在[data2vec]里下载了【data2vec ViT-B | 800 epochs pretrained】。并在train.py中通过“model.load_param_finetune(cfg.MODEL.PRETRAIN_PATH)”加载预训练文件想要进行微调。但是报关键字对应不上的错。请问这个预训练文件是您在finegpr数据集上跑完的预训练权重文件吗？

JeremyXSC commented 1 week ago

首先非常感谢你对我们工作的关注，我们这个模型需要用FineGPR数据集进行训练。