awesome-DeepLearning icon indicating copy to clipboard operation
awesome-DeepLearning copied to clipboard

训练时,如何只保存准确率最高的模型和最后一个模型final.pdparams

Open luyunfighting opened this issue 2 years ago • 3 comments

我只想保存这两个模型,不然太占空间了

luyunfighting avatar Sep 02 '21 02:09 luyunfighting

我用的是paddle.Model的高级API训练的,现在只能按save_freq保存

luyunfighting avatar Sep 02 '21 02:09 luyunfighting

https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/callbacks/ModelCheckpoint_cn.html#modelcheckpoint 我查了一下,有modelcheckpoint,ModelCheckpoint 回调类和model.fit联合使用,在训练阶段,保存模型权重和优化器状态信息。当前仅支持在固定的epoch间隔保存模型,不支持按照batch的间隔保存。可以把save_freq设置大一点。

https://github.com/PaddlePaddle/Paddle 建议您向框架api反馈一下这个问题哈

w5688414 avatar Sep 07 '21 02:09 w5688414

好的收到,感谢

luyunfighting avatar Sep 07 '21 07:09 luyunfighting