PyTorch预训练模型保存与加载

模型的保存

'''
torch.save()函数保存序列化的对象。
'''

# 保存整个模型
torch.save(model, './path')

# 仅保存参数
torch.save(model.state_dict(), './path')

模型的加载

# 整个模型的加载
model