PaddleOCR icon indicating copy to clipboard operation
PaddleOCR copied to clipboard

在model.train()情况下仅仅运行正向推理就会改变model?

Open sniperwrb opened this issue 1 year ago • 0 comments

请提供下述完整信息以便快速定位问题/Please provide the following information to quickly locate the problem

  • 系统环境/System Environment:linux
  • 版本号/Version:Paddle:2.6.0.post120
  • PaddleOCR:2.7.0.3
  • 问题相关组件/Related components:PaddleOCR-release-2.7/tools/program.py,def train(),模型是多语言模型中的latin的训练模型,SVTR那个
  • 运行指令/Command Code:从tools/train.py调用了train(),而且把下面这三步都注释掉了
avg_loss.backward()
optimizer.step()
optimizer.clear_grad()
  • 完整报错/Complete Error Message:没有报错,就是跑了1个epoch之后,模型的推理能力变得很差。 按理说应该是完全不变的对吧?实际上在model加载后直接save_model,或者把model.train()改成eval()跑一个epoch,结果也确实没问题。

sniperwrb avatar Feb 29 '24 07:02 sniperwrb