ADDA pytorch 代码实现

1 导入MNIST 和UPSP数据集：

从torchvision.datasets.MNIST下载即可，代码在datasets文件夹里mnist.py 和usps.py

下载后的MNIST图像都是灰度图像，只有一个通道。所以运行原来的程序会报错：

RuntimeError: output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]。

原来程序：

原来程序

报错原因：这是因为mnist图像都是灰度图像，只有一个通道，而上面的transforms.Normalize 却对三个通道都归一化了，这肯定会报错，所以只要像下面修改即可：

修改后