求助 训练好的模型该怎么保存
收藏
项目本身用的是torch.save()，将训练好的模型保存到本地，但是项目迁移到ai studio之后就不能存了
萌新第一次用ai studio跑项目 一般遇到这种模型保存的问题该怎么处理？
0
收藏
请登录后评论
paddle 1.8 的话，有save_params方法。如果是保存用于推理的模型的话，还有save_inference_model。
具体用法可以查一下API文档。
查一下API?
2.0rc 用paddle.save
paddle模型分persistable和inference两种
paddle.save()存的是persistable类型了模型，一般用于继续训练
paddle.jit.save()存储的是inference类型的模型，一般用于部署、调优
详情参考文档
https://www.paddlepaddle.org.cn/documentation/docs/zh/2.0-rc/guides/02_paddle2.0_develop/08_model_save_load_cn.html#id2
https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/save_cn.html