File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -93,7 +93,7 @@ The ``prepare_model_data.py`` script downloads the CIFAR-10 dataset in pickle fo
93
93
dummy_inputs,
94
94
tmp_model_path,
95
95
export_params=True,
96
- opset_version=13 ,
96
+ opset_version=17 ,
97
97
input_names=input_names,
98
98
output_names=output_names,
99
99
dynamic_axes=dynamic_axes,
@@ -102,7 +102,7 @@ The ``prepare_model_data.py`` script downloads the CIFAR-10 dataset in pickle fo
102
102
Note the following settings for the onnx conversion:
103
103
104
104
- Ryzen AI supports a batch size=1, so dummy input is fixed to a batch_size =1 during model conversion
105
- - Recommended ``opset_version `` setting 13 is used.
105
+ - Recommended ``opset_version `` setting 17 is used.
106
106
107
107
Run the following command to prepare the dataset and export the ONNX model:
108
108
You can’t perform that action at this time.
0 commit comments