refine bridge part of unet