pytorch如何将bin格式模型导出pt格式模型?

PyTorch提供了两种导出模型的方法:

  1. 将模型保存为.pt文件

用以下代码读取bin模型文件

python 复制代码
# 获取模型
model = WhisperForConditionalGeneration.from_pretrained(args.model_path,
                                                        device_map="auto",
                                                        local_files_only=args.local_files_only).half()

可以使用以下函数将模型保存为.pt文件:

python 复制代码
torch.save(model.state_dict(), path_to_file)

其中,model是要保存的模型,state_dict()是将模型中所有参数的值保存为一个字典,path_to_file是保存路径和文件名。

  1. 将模型保存为ONNX格式

可以使用以下函数将PyTorch模型保存为ONNX格式:

torch.onnx.export(model, input, path_to_file)

其中,model是要保存的模型,input是一个PyTorch张量,用于指定输入张量的形状和数据类型,path_to_file是保存路径和文件名。该函数将自动将模型转换为ONNX格式并保存到本地文件中。

需要说明的是,保存为.ONNX格式的模型可以被其他深度学习框架加载和使用。

相关推荐
3gying28 分钟前
chromedriver
python
DES 仿真实践家1 小时前
【Day 11-N22】Python类(3)——Python的继承性、多继承、方法重写
开发语言·笔记·python
云泽野6 小时前
【Java|集合类】list遍历的6种方式
java·python·list
IMPYLH7 小时前
Python 的内置函数 reversed
笔记·python
小赖同学啊9 小时前
物联网数据安全区块链服务
开发语言·python·区块链
码荼10 小时前
学习开发之hashmap
java·python·学习·哈希算法·个人开发·小白学开发·不花钱不花时间crud
小陈phd11 小时前
李宏毅机器学习笔记——梯度下降法
人工智能·python·机器学习
喝过期的拉菲11 小时前
如何使用 Pytorch Lightning 启用早停机制
pytorch·lightning·早停机制
kk爱闹11 小时前
【挑战14天学完python和pytorch】- day01
android·pytorch·python
Blossom.11811 小时前
机器学习在智能建筑中的应用:能源管理与环境优化
人工智能·python·深度学习·神经网络·机器学习·机器人·sklearn