Perfect, thanks plenty. When implementing the concatenation of AdaptiveAvgPool and AdaptiveMaxPool the loss hugely varies from epoch to epoch and does not decrease constantly. Do you know why this happens?
Another thing is that the loss is still quite high when using the 320x320 pixel images as inputs. Do you have an idea how to improve this?