logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

pytorch 模型保存和断点恢复训练

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

文章图片
#pytorch#人工智能#python
COCO2017数据集,mini coco数据集分享

链接:https://pan.baidu.com/s/17wpMzJvzSQ-a3qbaqIbmdg?

文章图片
#python#pytorch#人工智能 +2
到底了