I’m using this as last layer
self.final = nn.Sequential(
nn.MaxPool2d((100,100)),
nn.Sigmoid()
)
I had the feeling that there was something wrong at the output, so I checked manually the maximum value it is the same but when applying the sigmoid the results are different.
the input is of size
(64,1,100,100)