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 小时前
Python的Optional:让你的代码优雅处理“空值”危机
python
曾几何时`3 小时前
基于python和neo4j构建知识图谱医药问答系统
python·知识图谱·neo4j
石迹耿千秋5 小时前
迁移学习--基于torchvision中VGG16模型的实战
人工智能·pytorch·机器学习·迁移学习
写写闲篇儿6 小时前
Python+MongoDB高效开发组合
linux·python·mongodb
杭州杭州杭州7 小时前
Python笔记
开发语言·笔记·python
路人蛃8 小时前
通过国内扣子(Coze)搭建智能体并接入discord机器人
人工智能·python·ubuntu·ai·aigc·个人开发
qiqiqi(^_×)9 小时前
卡在“pycharm正在创建帮助程序目录”
ide·python·pycharm
Ching·9 小时前
esp32使用ESP-IDF在Linux下的升级步骤,和遇到的坑Traceback (most recent call last):,及解决
linux·python·esp32·esp_idf升级
吗喽15434518810 小时前
用python实现自动化布尔盲注
数据库·python·自动化
hbrown10 小时前
Flask+LayUI开发手记(十一):选项集合的数据库扩展类
前端·数据库·python·layui