search

Zj-BinXia / DiffIR

This project is the official implementation of 'Diffir: Efficient diffusion model for image restoration', ICCV2023
Apache License 2.0
451 stars 18 forks source link

终端显示'DiffIRS1' object has no attribute 'module' #65

Closed Justified4 closed 3 weeks ago

Justified4 commented 1 month ago

进行train_DiffIRS2_GANv2.sh时报错 Traceback (most recent call last): File "DiffIR/train.py", line 15, in train_pipeline(root_path) File "/root/autodl-tmp/DiffIR/train_pipeline.py", line 125, in train_pipeline model = build_model(opt) File "/root/miniconda3/lib/python3.8/site-packages/basicsr/models/init.py", line 26, in build_model model = MODEL_REGISTRY.get(opt['model_type'])(opt) File "/root/autodl-tmp/DiffIR/models/DiffIR_GAN_S2_model.py", line 44, in init self.model_Es1 = self.net_g_S1.module.E File "/root/miniconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1185, in getattr raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'DiffIRS1' object has no attribute 'module'

lq5203344 commented 12 hours ago

Have you solved this problem? I encountered the same problem.