PaddleHub icon indicating copy to clipboard operation
PaddleHub copied to clipboard

Trainer的checkpoint参数失败。

Open icm-ai opened this issue 3 years ago • 4 comments

Trainer中设置了checkpoint目录,但是在aistudio中进行fine-tune之后,并没有出现模型参数,如best_model之类的文件与文件夹在我设置的checkpoint目录下,尝试了好几次,设置过不同的目录,不是写权限问题,也尝试过默认选项。

在留下您的问题时,辛苦您同步提供如下信息:

  • 版本、环境信息 1)PaddleHub和PaddlePaddle版本:PaddleHub2.2.0与2.0.4都试过,PaddlePaddle2.1.2 2)系统环境:尝试过aistudio的 image 也尝试过最新的2.2的版本,都没有结果,目录中只有ckpt_1648390801.7935858/visualization出现,有一个log文件,看样子是可视化数据信息,如下。 image

  • 复现信息:没有报错,只是没有参数保存下来,无法进行fine-tune之后,下一步的预测,也就无法进行后续的部署之类的任务。

在train完成之后,输出日志如下 image 没有提示说参数保存之类的,也没有metric等信息,准确率auc之类的。

icm-ai avatar Mar 27 '22 15:03 icm-ai

您好,具体这个项目的链接是哪个,你给的这个可能是因为没权限访问,只能看到aistudio的主页,可以给一下你fork的最原始的那个项目链接

rainyfly avatar Mar 29 '22 06:03 rainyfly

https://aistudio.baidu.com/aistudio/projectdetail/3692580?shared=1 你试一下这个?不是fork的

icm-ai avatar Mar 29 '22 07:03 icm-ai

trainer.train里面有一个save_interval参数,默认是10,这个例子里你只跑了3个epoch,把epoch数量增大,或者把save_interval参数设置为1你再试试

rainyfly avatar Mar 29 '22 08:03 rainyfly

trainer.train里面有一个save_interval参数,默认是10,这个例子里你只跑了3个epoch,把epoch数量增大,或者把save_interval参数设置为1你再试试

我尝试把save_interval设置到1之后,checkpoint目录如愿以偿的出现了 image 非常感谢~

icm-ai avatar Mar 29 '22 09:03 icm-ai