pytorch卷积bug解决


class CNNNet(nn.Module):
    def __init__(self):
        super(CNNNet,self).__init__()
        self.layer1=nn.Sequential(
            nn.Conv2d(1,16,kernel_size=3),
            nn.ReLU(inplace=True)
            )

RuntimeError: Given groups=1, weight of size [16, 1, 3, 3], expected input[64, 227, 227, 3] to have 1 channels, but got 227 channels instead

 

 

兄弟你这个问题解决了吗?