pytorch保存模型方法

Pytorch 有两种保存模型的方式，都是通过调用pickle序列化方法实现的。

第一种方法只保存模型参数。第二种方法保存完整模型。推荐使用第一种，第二种方法可能在切换设备和目录的时候出现各种问题。

1.保存模型参数方法:

print(model.state_dict().keys())                                # 输出模型参数名称# 保存模型参数到路径"./data/model_parameter.pkl"
torch.save(model.state_dict(), "./data/model_parameter.pkl")
new_model = Model()                                                    # 调用模型Model
new_model.load_state_dict(torch.load("./data/model_parameter.pkl"))    # 加载模型参数
new_model.forward(input)                                               # 进行使用

2.保存完整模型(不推荐)

torch.save(model, './data/model.pkl')        # 保存整个模型
new_model = torch.load('./data/model.pkl')   # 加载模型

3.Transfomers库预训练模型的加载

# 使用transformers预训练后进行保存
model.save_pretrained(model_path)
tokenizer.save_pretrained(tokenizer_path)# 预训练模型使用 `from_pretrained()` 重新加载
model.from_pretrained(model_path)
tokenizer.from_pretrained(tokenizer_path)

pytorch保存模型方法相关推荐

pytorch保存模型pth_Day159:模型的保存与加载
网络结构和参数可以分开的保存和加载,因此,pytorch保存模型有两种方法: 保存整个模型 (结构+参数) 只保存模型参数(官方推荐) # 保存整个网络torch.save(model, check ...
PyTorch 保存模型结构参数及加载模型
PyTorch 保存模型结构参数及加载模型保存模型与加载保存模型分为两种方式: 保存整个网络结构和参数保存整个网络的参数 # 1.保存并加载整个网络结构和参数 # 保存模型 torch.save ...
pytorch保存模型的两种方法
文章目录前言一.保存整个模型二.只保存参数模型不同后缀名的区别总结前言模型的本质是一堆用某种结构存储起来的参数用数据对模型进行训练后得到了比较理想的模型,就需要将其存储起来,然后在需要 ...
pytorch保存模型pth_Pytorch_trick_04
科技猛兽:PyTorch 50.Pytorch模型保存与加载,并在加载的模型基础上继续训练zhuanlan.zhihu.com 1.Pytorch 模型保存与加载,并在加载的模型基础上继续训练只保 ...
pytorch保存模型pth_浅谈pytorch 模型 .pt, .pth, .pkl的区别及模型保存方式
我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件,这几种模型文件在格式上有什么区别吗? 其实它们并不是在格式上有区别,只是后缀不同而已(仅此而已),在用torch.save ...
pytorch保存模型pth_pytorch中保存的模型文件.pth深入解析
前言:前面有专门的讲解关于如何深入查询模型的参数信息,可以参考这篇文章: 沈鹏燕:pytorch教程之nn.Module类详解zhuanlan.zhihu.com 本次来解析一下我们通常保存的模型文 ...
[PyTorch] 保存模型和恢复模型
第一种:(推荐)只保存和加载模型参数保存模型 torch.save(the_model.state_dict(), PATH) 加载模型参数 the_model = TheModelClass(*a ...
pytorch保存模型pth_Pytorch模型的加载和保存是否有坑？
谢邀,我也不是很确定,手边暂时也没法google,先抛个砖吧,期待后续有大佬来解答. 之前用keras训练模型也遇到过类似的问题,pre-trained model 直接inference倒是没什么问 ...
pytorch保存模型运行时状态，记录点checkpoint
文章目录保存加载案例保存 torch.save({'epoch': epoch, # 保存迭代次数'model_state_dict': model.state_dict(), # 模型的状态 ...

pytorch保存模型方法

pytorch保存模型方法相关推荐

最新文章

热门文章