'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!