Does the training phase only require training reference unet？
Does the training phase only require training reference unet？