convert to ONNX

[fardinadii]

Hi, I'm trying to convert the model that I trained with a custom dataset to ONNX. First of all, I train the model and after that, I try to convert the model to ONNX using torch.onnx.export but I was faced with an Error. after that, I found that your code has something for converting the model to onnx (onnx_export).

I'm a little confused for onnx_export ... if we set onnx_export=True the final model is in ONNX format? could you please tell me how this onnx_export works?