for the frist test - nas-unet