Browse Source
opset_version to 14
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
timm_image.py
|
@ -208,7 +208,7 @@ class TimmImage(NNOperator): |
|
|
path, |
|
|
path, |
|
|
input_names=['input_0'], |
|
|
input_names=['input_0'], |
|
|
output_names=['output_0'], |
|
|
output_names=['output_0'], |
|
|
opset_version=12, |
|
|
|
|
|
|
|
|
opset_version=14, |
|
|
dynamic_axes={ |
|
|
dynamic_axes={ |
|
|
'input_0': {0: 'batch_size'}, |
|
|
'input_0': {0: 'batch_size'}, |
|
|
'output_0': {0: 'batch_size'} |
|
|
'output_0': {0: 'batch_size'} |
|
|