美文网首页
ADDA pytorch 代码实现

ADDA pytorch 代码实现

作者: 深度学习努力中 | 来源:发表于2019-06-11 15:50 被阅读0次

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 却对三个通道都归一化了,这肯定会报错,所以只要像下面修改即可:

修改后

相关文章

网友评论

      本文标题:ADDA pytorch 代码实现

      本文链接:https://www.haomeiwen.com/subject/dulsxctx.html