RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ... 错误解决方法
RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ... 这个问题是我在使用pytorch训练模型时遇到的,意思是5这个参数是无效的,我一开始认为是我数据集的问题,应该不是图片通道的问题,因为我用的都是三通道的,应该是图片尺寸不统一,就想着把它们都改成同样大小。在我改着的时候我突然意识...
原创
2019-10-24 14:25:03 ·
6150 阅读 ·
8 评论