计算机视觉与深度学习

Input and parameter tensors are not at the same device 出现这个问题的原因是没有把模型和数据都放到cuda里面 device torch.device(cuda:0 if torch.cu