UNET# the max pool can be at the begging - faster for smaller modern model can skip the maxPool and do stride*2;