Zj-BinXia/DiffIR

'DiffIRGANS2Model' object has no attribute 'model_Es1'

Closed this issue · 3 comments

Thank you for your excellent work!

When trying the upsampling factor is 2 and loading the pre-trained model, I met the following problem.
Traceback (most recent call last): File "DiffIR/train.py", line 15, in <module> train_pipeline(root_path) File "/DiffIR/DiffIR-SRGAN/DiffIR/train_pipeline.py", line 185, in train_pipeline model.optimize_parameters(current_iter) File "/DiffIR/DiffIR-SRGAN/DiffIR/models/DiffIR_GAN_S2_model.py", line 116, in optimize_parameters _, S1_IPR = self.model_Es1(self.lq,self.gt)
I would appreciate it if you could kindly give me some advice about this.

Thanks in advance.

Thank you for your interest. I have already updated the DiffIR_GAN_S2_model.py file to fix this bug.

Appreciate your reply! I'll confirm later, if this works in my program and updates will be shared promptly :)

Works! Thanks again for your help!