首页 Paddle框架 帖子详情
小白在使用模型训练时出现,KeyError: 'LinearWarmup_LR'
收藏
快速回复
Paddle框架 问答模型训练 763 1
小白在使用模型训练时出现,KeyError: 'LinearWarmup_LR'
收藏
快速回复
Paddle框架 问答模型训练 763 1

在自己的电脑上下载了ch_PP-OCRv2_rec_slim训练模型,把训练和测试图片和标注路径修改了,然后使用python tools/train.py -c configs/rec/ch_PP-OCRv2_rec.yml -o Global.checkpoints=./output/rec_mobile_pp-OCRv2/best_accuracy命令,接着跳出这个问题,请问是怎么回事呢?

0
收藏
回复
全部评论(1)
时间顺序
A
AIStudio2082122
#2 回复于2022-04

以下是问题上面的提示代码

 

Traceback (most recent call last):
File "E:\softwares\anaconda\PaddleOCR-release-2.4\PaddleOCR-release-2.4\tools\train.py", line 148, in
main(config, device, logger, vdl_writer)
File "E:\softwares\anaconda\PaddleOCR-release-2.4\PaddleOCR-release-2.4\tools\train.py", line 100, in main
pre_best_model_dict = load_model(config, model, optimizer)
File "E:\softwares\anaconda\PaddleOCR-release-2.4\PaddleOCR-release-2.4\ppocr\utils\save_load.py", line 83, in load_model
optimizer.set_state_dict(optim_dict)
File "E:\softwares\anaconda\lib\site-packages\decorator.py", line 232, in fun
return caller(func, *(extras + args), **kw)
File "E:\softwares\anaconda\lib\site-packages\paddle\fluid\wrapped_decorator.py", line 25, in __impl__
return wrapped_func(*args, **kwargs)
File "E:\softwares\anaconda\lib\site-packages\paddle\fluid\framework.py", line 229, in __impl__
return func(*args, **kwargs)
File "E:\softwares\anaconda\lib\site-packages\paddle\optimizer\optimizer.py", line 287, in set_state_dict
self._learning_rate.set_state_dict(state_dict["LR_Scheduler"])
File "E:\softwares\anaconda\lib\site-packages\paddle\optimizer\lr.py", line 794, in set_state_dict
self.learning_rate.set_state_dict(state_dict["LinearWarmup_LR"])

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户