建议创建项目的时候检查名称
Jinnrry opened this issue · comments
Jinnrry commented
建议执行python app.py create {project_name}
命令的时候,检查一下project_name
中是否包含下划线,包含下划线的时候抛个错误,不要让创建了
原因:
https://github.com/sml2h3/dddd_trainer/blob/main/utils/train.py#L58
在这个地方加载checkpoints的时候,使用下划线分割文件名,如果自己项目名里面包含下划线,那么这里将加载失败。最终就会像我这里,训练了几天的模型,再次训练的时候加载不成功
Jinnrry commented
提了个PR,在取模型训练步数的时候用分割后的数组的最后一个值,这样项目名包含下划线也没问题了