ONNX export opset version

[huettern]

When doing the first nemo.utils.export_onnx in the MNIST example, I get an error

RuntimeError: Exporting the operator round to ONNX opset version 9 is not supported. Support for this operator was added in version 11, try exporting with this version.

Appending opset_version=11 to the utils.py file torch.onnx.export fixed the issue for me. Is that the correct fix?

[FrancescoConti]

Hi @Noah95 , sorry for the delay in the response. What version of PyTorch and of NEMO are you using?

[huettern]

No worries, these are the versions:

pytorch-nemo==0.0.5
torch==1.5.0

[FrancescoConti]

Ok I had a rapid look... I could not reproduce the error so far. Can you share more details on the network you are exporting, whether you already quantized it, etc?