refactor(train): 注释模型导出相关代码

- 注释掉了 opset_version 和 simplify 参数
- 保留了 format 和 dynamic 参数
This commit is contained in:
fly6516 2025-05-29 21:06:56 +08:00
parent 21d190a658
commit 86ccd7c29e

View File

@ -111,11 +111,11 @@ def main():
# === 模型导出 ===
print("\n=== 模型导出 ===")
model.export(
opset_version=15,
# opset_version=15,
format="onnx", # 优先导出ONNX
# format="torchscript",
dynamic=True, # 动态维度
simplify=True # 简化模型
# simplify=True # 简化模型
)
if __name__ == "__main__":