江南烟雨尘

摘要： 1.测试时不需要梯度： with torch.no_grad(): out = net(input) 2.可视化网络生成的图片（整个batch） from torchvision.utils import save_image out = net(input) save_image(out,'./a 阅读全文