From 4027afe1219d35de29ee652c2c6472360a19657d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Fri, 30 Aug 2024 11:13:41 +0200
Subject: [PATCH] fix : upped the torh export to onnx opset from 11 to 15 to
remove a warning
---
aidge_interop_torch/utils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/aidge_interop_torch/utils.py b/aidge_interop_torch/utils.py
index 4b59251..9e639ac 100644
--- a/aidge_interop_torch/utils.py
+++ b/aidge_interop_torch/utils.py
@@ -261,7 +261,7 @@ class ContextNoBatchNormFuse:
def wrap(torch_model: torch.nn.Module,
input_size: Union[list, tuple],
- opset_version: int = 11,
+ opset_version: int = 15,
in_names: list = None,
out_names: list = None,
verbose: bool = False) -> AidgeModule:
--
GitLab