Pytorch 保存中途训练的模型

一般看到的是这个：

torch.save(net.state_dict(), 'model.pth')

但这个只保存了模型本身的参数，能保证可以load保存的模型进行测试，但是并不方便恢复训练。

实际上还需要保存优化器的状态，以及一些其他有助于恢复训练的信息(如loss等)：

def save_checkpoint(epoch, epochs_since_improvement, model, optimizer, loss, is_best):state = {'epoch': epoch,'epochs_since_improvement': epochs_since_improvement,'loss': loss,'model': model,'optimizer': optimizer}filename = 'checkpoint_' + str(epoch) + '_' + str(loss) + '.tar'torch.save(state, filename)if is_best:torch.save(state, 'BEST_checkpoint.tar')

Pytorch 保存中途训练的模型相关推荐

pt-＞onnx-＞ncnn（pytorch部署自己训练的模型）
pt->onnx->ncnn(pytorch部署自己训练的模型) yolov6似乎有部分操作ncnn不支持,需要改一下网络结构,所以这里使用 yolov7-tiny 首先,找一个目标检测的 ...
pytorch 保存、加载模型
一般保存为.pt格式,保存模型使用: torch.save(model, '保存位置') 加载模型使用: model_load = torch.load('加载模型的位置') 完整代码 import ...
关于使用Pytorch时，训练集模型表现很好但测试集模型表现极差的原因
出现这一现象的原因主要有三个: 训练数据集过小,导致系统泛化能力不足训练和测试时的model.train(), model.eval()没有设置好训练时数据集一定要打乱,Dataloader的sh ...
PyTorch框架学习十九——模型加载与保存
PyTorch框架学习十九--模型加载与保存一.序列化与反序列化二.PyTorch中的序列化与反序列化 1.torch.save 2.torch.load 三.模型的保存 1.方法一:保存整个Mo ...
pytorch 驱动不兼容_解决Pytorch 加载训练好的模型遇到的error问题
这是一个非常愚蠢的错误 debug的时候要好好看error信息提醒自己切记好好对待error!切记!切记! -----------------------分割线---------------- py ...
python torch exp_Python:PyTorch 保存和加载训练过的网络 (八十)
保存和加载模型在这个 notebook 中,我将为你展示如何使用 Pytorch 来保存和加载模型.这个步骤十分重要,因为你一定希望能够加载预先训练好的模型来进行预测,或是根据新数据继续训练. %m ...
pytorch保存模型pth_Day159:模型的保存与加载
网络结构和参数可以分开的保存和加载,因此,pytorch保存模型有两种方法: 保存整个模型 (结构+参数) 只保存模型参数(官方推荐) # 保存整个网络torch.save(model, check ...
保存模型后无法训练_模型构建到部署实践
导读在工业界一般会采用了tensorflow-serving进行模型的部署,而在模型构建时会因人而异会使用不同的深度学习框架,这就需要在使用指定深度学习框架训练出模型后,统一将模型转为pb格式,便于 ...
Pytorch基础训练库Pytorch-Base-Trainer(支持模型剪枝分布式训练)
Pytorch基础训练库Pytorch-Base-Trainer(支持模型剪枝分布式训练) 目录 Pytorch基础训练库Pytorch-Base-Trainer(PBT)(支持分布式训练) 1.I ...

Pytorch 保存中途训练的模型

Pytorch 保存中途训练的模型相关推荐

最新文章

热门文章