[CI] Upgrade pytorch to 2.4.1 (#784)

* upgrade pytorch to 2.4.1 * use onnx opset 13
microsoft · Oct 16, 2024 · c5186d8 · c5186d8
1 parent ccc3ada
commit c5186d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml
@@ -32,7 +32,7 @@ jobs:
         python-version: ['3.9', '3.10', '3.11']
     env:
       TVM_VERSION_TAG: v0.17.0
-      PYTORCH_VERSION: 2.2.0
+      PYTORCH_VERSION: 2.4.1
       LLVM_VERSION: 18
 
     steps:

diff --git a/hummingbird/ml/_topology.py b/hummingbird/ml/_topology.py
@@ -243,7 +243,7 @@ def convert(topology, backend, test_input, device, extra_config={}):
 
     if backend == onnx.__name__:
         onnx_model_name = output_model_name = None
-        target_opset = 11
+        target_opset = 13
 
         # Set optional configuration options for ONNX if any.
         if constants.ONNX_OUTPUT_MODEL_NAME in extra_config: