diff --git a/train.py b/train.py index 57d6011..81b6e11 100644 --- a/train.py +++ b/train.py @@ -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__":