tencent-ailab / IP-Adapter

The image prompt adapter is designed to enable a pretrained text-to-image diffusion model to generate images with image prompt.
Apache License 2.0
4.46k stars 289 forks source link

is it right while unet set in evaluation mode for training? #370

Open rardz opened 1 month ago

rardz commented 1 month ago

from thetutorial_train.py, after unet is loaded and not specificly set to training mode. is it right for training ip-adapter parameter in evaluation mode?

xiaohu2015 commented 1 month ago

from thetutorial_train.py, after unet is loaded and not specificly set to training mode. is it right for training ip-adapter parameter in evaluation mode?

hi, the 'train' and 'val' mode have same behavior for UNet (don't have dropout and BN)