ICPR_TextDection icon indicating copy to clipboard operation
ICPR_TextDection copied to clipboard

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

Open huahuabai opened this issue 3 years ago • 1 comments

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

huahuabai avatar Jan 26 '22 13:01 huahuabai

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

UpCoder avatar Jan 27 '22 04:01 UpCoder