Mobile bert : onnxruntime :: training :: GradientBuilderBase :: O i < node_ -> Outputdefs().size was false

[Leaner23]

I am trying to generate the training artifacts for mobile bert model but onnx is giving me below error message : : onnxruntime :: training :: GradientBuilderBase :: O i < node_ -> Outputdefs().size was false.

I am trying the example that is given under masked_language_modeling and the name of file is mobilebert_offline.ipynb.

Please help !! It is urgent!!

[prathikr]

@carzh related issue: microsoft/onnxruntime#19038

[carzh]

Hi @Leaner23 , adding the following line: opset_version=15, to the torch.onnx.export parameters should fix the issue. Thanks to @prathikr for finding the issue & solution.

PR for fix is here #179

[carzh]

Closing since PR for fix has been merged into master -- please reopen if you run into related issues!!