
简介
该用户还未填写简介
擅长的技术栈
未填写擅长的技术栈
可提供的服务
暂无可提供的服务
pytorch 模型保存和断点恢复训练
在使用pytorch训练模型中,有时候需要从断点处继续训练,那么需要将模型、优化器、lr_scheduler,epoch和其他args进行保存,如果使用的apm混合精度,还要保存scaler。在恢复训练的时候,需要重新加载数据即可,在推理阶段,只要保存model.state_dict()就好了。

COCO2017数据集,mini coco数据集分享
链接:https://pan.baidu.com/s/17wpMzJvzSQ-a3qbaqIbmdg?

到底了







