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

[ney158]

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]

I haven't encountered this problem before

[yanmengyi]

I have the same problem

[d-linlin]

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

[Xia-zx]

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

[So-cean]

same problem

[So-cean]

using

import os.path as osp
import sys
sys.path.append(osp.abspath(osp.join(__file__, osp.pardir, osp.pardir)))

from basicsr.test import test_pipeline
import DiffIR.archs
import DiffIR.data
import DiffIR.models

if __name__ == '__main__':
    root_path = osp.abspath(osp.join(__file__, osp.pardir, osp.pardir))
    print(root_path)
    test_pipeline(root_path)

in test.py