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

相关推荐
万邦科技Lafite4 小时前
京东按图搜索京东商品(拍立淘) API (.jd.item_search_img)快速抓取数据
开发语言·前端·数据库·python·电商开放平台·京东开放平台
丁浩6664 小时前
Python机器学习---6.集成学习与随机森林
python·随机森林·机器学习
charlie1145141915 小时前
现代 Python 学习笔记:Statements & Syntax
笔记·python·学习·教程·基础·现代python·python3.13
麦麦大数据8 小时前
F036 vue+flask中医热性药知识图谱可视化系统vue+flask+echarts+mysql
vue.js·python·mysql·flask·可视化·中医中药
移远通信8 小时前
MQTT协议:物联网时代的通信革命
python·物联网·网络协议
Amo Xiang8 小时前
JavaScript逆向与爬虫实战——基础篇(css反爬之动态字体实现原理及绕过)
爬虫·python·js逆向·动态字体
编程让世界美好9 小时前
选手评分问题(python)
python
java1234_小锋9 小时前
PyTorch2 Python深度学习 - PyTorch2安装与环境配置
开发语言·python·深度学习·pytorch2
CClaris9 小时前
深度学习——反向传播的本质
人工智能·python·深度学习
伊玛目的门徒9 小时前
Jupyter Notebook 配置使用虚拟环境中(virtualenv) 内核
python·jupyter·virtualenv