sml2h3 / dddd_trainer

ddddocr训练工具

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

建议创建项目的时候检查名称

Jinnrry opened this issue · comments

建议执行python app.py create {project_name}命令的时候,检查一下project_name中是否包含下划线,包含下划线的时候抛个错误,不要让创建了

原因:
https://github.com/sml2h3/dddd_trainer/blob/main/utils/train.py#L58

在这个地方加载checkpoints的时候,使用下划线分割文件名,如果自己项目名里面包含下划线,那么这里将加载失败。最终就会像我这里,训练了几天的模型,再次训练的时候加载不成功

提了个PR,在取模型训练步数的时候用分割后的数组的最后一个值,这样项目名包含下划线也没问题了