RuntimeError: Unknown IValue type for pickling: Device

Question

i want convert pytorch model to torchscript, the model is a pytorch LSTM module.

step 1: score_pos is normal as expected