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
兄弟你这个问题解决了吗?