In model function "forward", after "out = F.avg_pool2d(out, 4)", need do 2d average pooling. Before this, out.size=[-1, 512, 7, 7],after this, out.size=[-1, 512, 1, 1].
define this in net:
self.pool2 = nn.AvgPool2d(7)
use it in function "forward":
out = F.avg_pool2d(out, 4)
out = self.pool2(out)
out = out.view(out.size(0), -1)
The strange thing is that I reported an error when verifying the model parameters, but no error was reported during training
In model function "forward", after "out = F.avg_pool2d(out, 4)", need do 2d average pooling. Before this, out.size=[-1, 512, 7, 7],after this, out.size=[-1, 512, 1, 1].
define this in net:
self.pool2 = nn.AvgPool2d(7)
use it in function "forward":
out = F.avg_pool2d(out, 4)
out = self.pool2(out)
out = out.view(out.size(0), -1)
The strange thing is that I reported an error when verifying the model parameters, but no error was reported during training