Fashion CNN基础教程:https://www.kaggle.com/pankajj/fashion-mnist-with-pytorch-93-accuracy/notebook
1.教程中灰度图片替换为RGB图片后遇到的问题:
Given groups=1, weight of size [32, 1, 3, 3], expected input[10000, 3, 32, 32] to have 1 channels, but got 3 channels instead
网上提供了很多方法,比如将RGB图片转换为灰度图(笑),或者在读入时选择convert(‘RGB') (读入方法不同所以没有尝试这种方法,但分析觉得治标不治本)
目前认为最有效的方法是将二维卷积层的输入通道数目从1增加到相应的数量(这里为3)(如下)。
不知道为什么没有找到相关记录,可能是进行到后面就自然解决了?
nn.Conv2d(in_channels=3, out_channels=32, kernel_size=3, padding=1)
2.CUDA显存不够
RuntimeError: CUDA out of memory. Tried to allocate 1.22 GiB (GPU 0; 2.00 GiB total capacity; 153.3