Can convert model to onnx but can't perform jit script or trace

Does torch.onnx use jit script/trace to perform its conversion?

Yes, as described here.

That is odd because for
jit trace and script I get

AssertionError: Expected Module but got <class ‘method’>

with torch.onnx.convert everything runs perfectly. When I read the doc you shared carefully, it states that the converter does something equivalent to jit trace.

Do you have any advice on debugging jit script/tracing?