More importantly, if I just keep the model on the CPU, it also gives the same error information “can’t convert CUDA tensor to numpy…” I do not know why it happens. I understand the meaning of the error information, but I do not know how to solve it. It is very strange.