deepglugs / deep_imagen

scripts for running and training imagen-pytorch
38 stars 8 forks source link

What should the pred_objectives for unet2? #16

Open zhaobingbingbing opened 1 year ago

zhaobingbingbing commented 1 year ago

Hi, I noticed the default setting of pred_objectives in imagen_pytorch is 'noise'. But in your imagenconfig, is that mean you set the pred_objectives as 'x_start' for unet2? Is there any special reason for it,can I just set it as 'noise' for unet2? image