Is there a list of models that can export ONNX?

[wxqwinner]

'vad_multilingual_frame_marblenet' can.t export. is there a restriction on exporting models for some reason?

import nemo.collections.asr as nemo_asr
model = nemo_asr.models.EncDecFrameClassificationModel.from_pretrained(model_name="vad_multilingual_frame_marblenet")

model.export('test.onnx', onnx_opset_version=12)

error:
'EncDecFrameClassificationModel' object has no attribute 'output_names'

[titu1994]

It depends on a case by case basis, not all models are exportable. We don't have a list as such of models that are Exportable, but generally they will subclass Exportable mixin.

@stevehuang52 @fayejf is this model supposed to be exportable ?

[stevehuang52]

sorry for the delay, I'll take a loot and update later. EncDecFrameClassificationModel inherits the EncDecClassificationModel class which was able to export some time ago, I'll check if something's changed..

[stevehuang52]

@wxqwinner The export error was fixed here 7076, please use the latest NeMo main branch and let use know if there're further issues.

[github-actions]

This issue is stale because it has been open for 30 days with no activity. Remove stale label or comment or this will be closed in 7 days.

[github-actions]

This issue was closed because it has been inactive for 7 days since being marked as stale.