Skip to content

Commit fa607e3

Browse files
committed
Update example in exporter docsrc for output_names usage
1 parent 282de22 commit fa607e3

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

docsrc/source/api/api_docs/modules/exporter.rst

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,22 @@ By default, the used ONNX opset version is 15, but this can be changed using `on
173173
174174
|
175175
176+
+++++++++++++++++++++++++++
177+
ONNX model output names
178+
+++++++++++++++++++++++++++
179+
180+
To set the model output names the argument `output_names` can be used:
181+
182+
.. code-block:: python
183+
184+
mct.exporter.pytorch_export_model(
185+
model=quantized_exportable_model,
186+
save_model_path=onnx_file_path,
187+
repr_dataset=representative_data_gen,
188+
output_names=['model_output'])
189+
190+
|
191+
176192
++++++++++++++++++++++++++++++++++++
177193
Use exported model for inference
178194
++++++++++++++++++++++++++++++++++++

0 commit comments

Comments
 (0)