ValueError: Expected more than 1 value per channel when training, got input size [1, 512, 1, 1]
个人碰到这个问题的时候是在缩放的时候,由于最后缩放大小只有1X1 了,我还有一个下采样代码,导致报了这个错
这里提供借鉴,或许可能,你也是这个错误
ValueError: Expected more than 1 value per channel when training, got input size [1, 512, 1, 1]
个人碰到这个问题的时候是在缩放的时候,由于最后缩放大小只有1X1 了,我还有一个下采样代码,导致报了这个错
这里提供借鉴,或许可能,你也是这个错误