2d Unet and 3d Unet average