Zj-BinXia / DiffIR

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

在windows上训练,报错No object named 'DiffIRS1Model' found in 'model' registry #42

Open ney158 opened 6 months ago

ney158 commented 6 months ago

File "DiffIR/train.py", line 12, in train_pipeline(root_path) File "D:\project\Super_resolution\DiffIR-SRGAN\DiffIR\train_pipeline.py", line 126, in train_pipeline model = build_model(opt) File "D:\Users\15822\Anaconda3\envs\pytorch\lib\site-packages\basicsr\models__init__.py", line 26, in build_model model = MODEL_REGISTRY.get(opt['model_type'])(opt) File "D:\Users\15822\Anaconda3\envs\pytorch\lib\site-packages\basicsr\utils\registry.py", line 71, in get raise KeyError(f"No object named '{name}' found in '{self._name}' registry!") KeyError: "No object named 'DiffIRS1Model' found in 'model' registry!"

Zj-BinXia commented 5 months ago

I haven't encountered this problem before

yanmengyi commented 4 months ago

I have the same problem

d-linlin commented 3 months ago

我遇到了同样的问题,请问你解决了吗

Xia-zx commented 2 months ago

试一下这个export PYTHONPATH=/path/to/your/project:$PYTHONPATH