self.conv1 = nn.Conv2d(1, 64, (1, 200), (1, 1), (0, 0))
self.batchnorm1 = nn.BatchNorm2d(64)
and, I try,
h = F.relu(self.batchnorm1(self.conv1(h)))
so, error occur as follow,
RuntimeError: running_mean should contain 64 elements not 128
I can’t understand why this error occur.
shape of conv1 is torch.Size([128, 64, 20, 1]) so, batch size is 128.
Please help.