Hello，how is your progress in trying to modify the JIT stuff?
I don’t think it is a long-term solution to solve the convert problem just by modify the batchfirst flag.because the origin pytorch model is not trained by myself.