Is ORTModule supporting only NLP models for now?

Hello,

Can someone please let me know if is this feasible to use ORTModule with Pytorch models like Deeplab/ Fcos, FasterRCNN or Maskrcnn?.

Examples and documentation only mentions NLP models. I however did try to wrap these models into an ORTModule but it did not work. At the begging the errors were related to missing operators that I had to register as custom operators but the last error error shown is: errors.SymbolicValueError( torch.onnx.errors.SymbolicValueError: ONNX export does NOT support exporting bitwise OR for non-boolean input values .

Should these wraps work? Or is it these Deep Learning models are still not supported for ortModule?

Appreciate your help,

Zazhil-ha