L1aoXingyu/pytorch-beginner

模型建立问题

kiddnesta opened this issue · 1 comments

model = Cnn(1, 10) #
不应该是Cnn(3,10)吗 rgb三通道

mnist数据集是灰度图,只有一个通道