RuntimeError #1
Description
运行第3个LeNet-5案例，自定义数据，最后模型测试出现这样的报错：RuntimeError: Given groups=1, weight of size [6, 1, 5, 5], expected input[1, 3, 32, 32] to have 1 channels, but got 3 channels instead
{{ message }}
运行第3个LeNet-5案例，自定义数据，最后模型测试出现这样的报错：RuntimeError: Given groups=1, weight of size [6, 1, 5, 5], expected input[1, 3, 32, 32] to have 1 channels, but got 3 channels instead