Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
在Pytorch上做QAT训练时,最开始使用的CPU,训练正常,后来尝试在GPU上训练,于是改动代码如下:#device=torch.device("cpu")device=torch.device("cuda:0"iftorch.cuda.is_available()else"cpu")在训练时出现题目中错误:Expectedalltensorstobeonthesamedevice,butfo