设备不一致问题

想问下大佬们，RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat1 in method wrapper_addmm)
这种设备不一致的报错该怎么搞呀，上网搜了好久都没解决，只知道是要找出哪些变量和计算在CPU上然后.to(device)到GPU上，但是我不知道怎么找，试过断点一步步运行了也找不到哪些张量是在cpu上的代码是GitHub上找的代码