ONNX版本降级

最近发现了onnx库中其实可以直接修改输出模型的format版本，代码如下：

import onnx

model = onnx.load(r"C:\Users\33132\Desktop\exp14\weights\SourceModelName.onnx")
model.ir_version = 6
onnx.save_model(model, r"C:\Users\33132\Desktop\ObjectModelName.onnx")

上述代码仅需准备好原始ONNX模型文件即可，然后写好路径后运行上述路径即可完成模型降级。

# python # 深度学习 # 机器学习
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

第6周学习总结：代码执行工具 + 多工具初步整合

本周完成了代码执行工具的安全沙箱设计，并为 Agent 添加了联网搜索能力，工具集扩展至 4 个，实现了从“只说不做”到“既说又做”的能力升级。

avatar 2048 AI社区
cover

OBS直播使用教程：OBS美颜插件OBS美颜摄像头OBS美颜相机下载安装使用教程

avatar 2048 AI社区

收藏！2026年小白程序员必入局的高薪AI赛道（含10大吃香岗位）

2026年AI岗位激增12倍，月薪破6万，供需比仅0.97，企业高薪抢人。文章介绍了10个前景广阔的AI岗位，如AI科学家、大模型算法工程师、AI产品经理等，部分岗位适合非计算机背景者。提供内部转型、转行过渡、自学上岸、考取证书等四条普通人入局AI的路径，强调越早入局机会越大。---

avatar 2048 AI社区