Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

请问该怎么将预训练模型微调到小数据集上呢,更改checkpoint的路径还是pretrained_model的路径呢 #8

Open
huahuabai opened this issue Jan 26, 2022 · 1 comment

Comments

@huahuabai
Copy link

请问该怎么将预训练模型微调到小数据集上呢,更改checkpoint的路径还是pretrained_model的路径呢?我更改了checkpoint的位置并且设置Restore为True如下
image
但是出现了如下问题
image
并没有对作者模型进行更改,请问您了解这是什么原因吗?盼复

@UpCoder
Copy link
Owner

UpCoder commented Jan 27, 2022

应该是通过指定 pretrained_model_path 来加载预训练模型,类似backbone啥的

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants