2022年10月28日
MNIST数据集output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
摘要： transform = transforms.Compose([ transforms.ToTensor(), transforms.Lambda(lambda x: x.repeat(3,1,1)), #添加这行 transforms.Normalize(mean=(0.5, 0.5, 0.5), 阅读全文
posted @ 2022-10-28 19:17 sdibt布谷鸟 阅读(258) 评论(0) 推荐(0) 编辑