awesome-DeepLearning
awesome-DeepLearning copied to clipboard
训练时,如何只保存准确率最高的模型和最后一个模型final.pdparams
我只想保存这两个模型,不然太占空间了
我用的是paddle.Model的高级API训练的,现在只能按save_freq保存
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反馈一下这个问题哈
好的收到,感谢