使用yml配置训练模型注意点
收藏
在用PaddleOCR做微调的时候，yml文件中有个参数 pretrain_weights 他的意思跟名字一样，即需要加载模型的参数
我在~/目录下有一个文件夹，暂且叫做 dir1，这个文件夹中有
best_accuracy.pdopt
best_accuracy.pdmodel
best_accuracy.pdparams
三个文件，既然要加载该模型，就需要指定路径。
开始我的参数设置成 pretrain_weights: /home/aistudio/dir1 他报错
后来把参数改成 pretrain_weights: /home/aistudio/dir1/best_accuracy 就能正常训练了
以前在写2.0动态图的时候也遇到这种情况，特地记录一下帮助有需要的人
1
收藏
能啊，txt形式打开
请问训练完成后，output里面只生成了.yml和.log这两个训练日志文件，而并没有生成.pdopt 和 .pdmodel 文件是怎么回事？
得有保存模型的代码，例如 model.save()