Could you check, if prob_interpolated has a valid .grad_fn attribute?
If print(prob_interpolated.grad_fn) returns None, the model output seems to be detached from the computation graph and this error would be raised.