First of all, thank you, but still gave an error.



 Yellow-Champagne:

Traceback (most recent call last):
File “D:/zgy/GCC/GCC-SFCN-master/train.py”, line 204, in <module>
main()
File “D:/zgy/GCC/GCC-SFCN-master/train.py”, line 81, in main
validate(val_loader, net, epoch, restore_transform)
File “D:/zgy/GCC/GCC-SFCN-master/train.py”, line 131, in validate
val_loss.append(net.loss.data.numpy())
TypeError:cannot convert CUDA tensor to numpy .Use Tensor.cpu() to copy the tensor to host memory first.