RuntimeError Given groups=1 weight of size 3 1 3 3 expected input1 3 8 8 to have 1 channels but got 3 channels instead报错
这个错误是因为在使用Conv2d卷积层时,设置了groups参数为1,表示输入和输出的通道数相同,但是输入的数据有3个通道,而权重只有1个通道,导致无法进行卷积操作。需要修改权重的通道数或者调整输入数据的通道数,使它们相同。
原文地址: http://www.cveoy.top/t/topic/bAmW 著作权归作者所有。请勿转载和采集!