Torchscript states device mismatch

I don’t know which model you are using, but maybe this error is related to this one which also failed if the model was traced but I believe scripting worked.

1 Like