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

- 注释掉了 opset_version 和 simplify 参数 - 保留了 format 和 dynamic 参数
2025-05-29 21:06:56 +08:00
parent 21d190a658
commit 86ccd7c29e
1 changed files with 2 additions and 2 deletions
@@ -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__":