Tensor for argument #2 ‘mat1‘ is on CPU, but expected it to be on GPU (while checking arguments for
Tensor for argument #2 ‘mat1’ is on CPU, but expected it to be on GPU (while checking arguments for addmm)需要将模型和输入数据都移动到 devicemodel=NonLinearRegression().to(device)#模型for batch_idx,(data,target) in enumerate(train_loader): data,target=data.to(device),