Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
If the forward parameter order does not match the tuple input order in jit.trace, like a text classification model, jit.trace will fail and we are capturing this with the exception here to make it fallback.