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)