Release 2.4.2

What's Changed

API changes

• Introduced the output_names argument in pytorch_export_model . This optional parameter specifies a list of output node names for export compatibility and is applicable only when using PytorchExportSerializationFormat.ONNX.
• Usage example: To set the model output names, use the output_names argument:

mct.exporter.pytorch_export_model(
    model=quantized_exportable_model,
    save_model_path=onnx_file_path,
    repr_dataset=representative_data_gen,
    output_names=['model_output'])

• Now the output name (i.e., the name of the output layer in the exported ONNX model) will be model_output, instead of the default name output, which would be assigned if output_names is not specified.