search

NJU-PCALab / AddSR

71 stars 2 forks source link

AttributeError: 'DDPMScheduler' object has no attribute 'get_z0' #3

Closed ziyizhetutanota closed 2 months ago

ziyizhetutanota commented 2 months ago

DDPM has no funtion get_z0, it seem to chnage the source codes

CSRuiXie commented 2 months ago

You can replace the original 'scheduling_ddpm.py' file with this, then it should work.

ziyizhetutanota commented 2 months ago

thank you. Another question, I'd like to ask that your discriminator was trained from scrach or iused a pre-trained model

CSRuiXie commented 2 months ago

We train the discriminator from scrach.

ziyizhetutanota commented 2 months ago

Thank you. Can you provide your pretrained discriminator ? I want to finetune my model.

CSRuiXie commented 2 months ago

We haven't saved the parameters of the discriminator model, so we can't provide the pre-trained discriminator. Sorry about that.

ziyizhetutanota commented 2 months ago

OK，thank you anyway