diff --git a/timm_image.py b/timm_image.py index e7f936e..3e9d260 100644 --- a/timm_image.py +++ b/timm_image.py @@ -208,7 +208,7 @@ class TimmImage(NNOperator): path, input_names=['input_0'], output_names=['output_0'], - opset_version=12, + opset_version=14, dynamic_axes={ 'input_0': {0: 'batch_size'}, 'output_0': {0: 'batch_size'}