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格式的模型可以被其他深度学习框架加载和使用。

相关推荐
这里有鱼汤1 小时前
miniQMT下载历史行情数据太慢怎么办?一招提速10倍!
前端·python
databook10 小时前
Manim实现脉冲闪烁特效
后端·python·动效
程序设计实验室11 小时前
2025年了,在 Django 之外,Python Web 框架还能怎么选?
python
倔强青铜三12 小时前
苦练Python第46天:文件写入与上下文管理器
人工智能·python·面试
用户25191624271116 小时前
Python之语言特点
python
刘立军16 小时前
使用pyHugeGraph查询HugeGraph图数据
python·graphql
数据智能老司机19 小时前
精通 Python 设计模式——创建型设计模式
python·设计模式·架构
数据智能老司机20 小时前
精通 Python 设计模式——SOLID 原则
python·设计模式·架构
c8i1 天前
django中的FBV 和 CBV
python·django
c8i1 天前
python中的闭包和装饰器
python