unable to trace the model using TorchScript

#3
by Iamexperimenting - opened

Hi team, I'm trying to trace this model using torchscript, but its not working. Its throwing IndexError: too many indices for tensor of dimension 2 error. Could you please tell me how to fix this for this model?

can someone help me here?

Sign up or log in to comment