Pytorch之保存读取模型实例

Pytorch之保存读取模型实例之保存读取模型实例

今天小编就为大家分享一篇Pytorch之保存读取模型实例，具有很好的参考价值，希望对大家有所帮助。一起跟

随小编过来看看吧

pytorch保存数据保存数据

pytorch保存数据的格式为.t7文件或者.pth文件，t7文件是沿用torch7中读取模型权重的方式。而pth文件是python中存储文件的

常用格式。而在keras中则是使用.h5文件。

# 保存模型示例代码

print('===> Saving models...')

torch.save(state, './checkpoint/autoencoder.t7')

保存用到torch.save函数，注意该函数第一个参数可以是单个值也可以是字典，字典可以存更多你要保存的参数（不仅仅是权

重数据）。

pytorch读取数据读取数据

pytorch读取数据使用的方法和我们平时使用预训练参数所用的方法是一样的，都是使用load_state_dict这个函数。

下方的代码和上方的保存代码可以搭配使用。下方的代码和上方的保存代码可以搭配使用。

print('===> Try resume from checkpoint')

if os.path.isdir('checkpoint'):

try:

checkpoint = torch.load('./checkpoint/autoencoder.t7')

start_epoch = 0

print('===> Start from scratch')

以上是pytorch读取的方法汇总，但是要注意，在使用官方的预处理模型进行读取时，一般使用的格式是pth，使用官方的模型

读取命令会检查你模型的格式是否正确，如果不是使用官方提供模型通过下面的函数强行读取模型(将其他模型例如caffe模型

转过来的模型放到指定目录下)会发生错误。

def vgg19(pretrained=False, **kwargs):

"""VGG 19-layer model (configuration "E")

Args:

pretrained (bool): If True, returns a model pre-trained on ImageNet

model_dir = '自己的模型地址'

model = VGG()

model.load_state_dict(torch.load(model_dir + 'vgg_conv.pth'))

也就是pytorch的读取函数进行读取即可。

以上这篇Pytorch之保存读取模型实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持我

们。