allennlp.predictors.predictor.get_gradients got error RuntimeError: Expected object of backend CUDA but got backend CPU for argument #3 'index'

Question

I tried to use allennlp.predictors.Predictor.get_gradients to get the gradients of an instance.

The main code:

for instance in targeted_dev_data:

predic