DiffIR icon indicating copy to clipboard operation
DiffIR copied to clipboard

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

Open ney158 opened this issue 2 years ago • 6 comments

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

ney158 avatar Mar 18 '24 10:03 ney158

I haven't encountered this problem before

Zj-BinXia avatar Apr 01 '24 11:04 Zj-BinXia

I have the same problem

yanmengyi avatar May 12 '24 10:05 yanmengyi

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

d-linlin avatar Jun 07 '24 02:06 d-linlin

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

Xia-zx avatar Jun 24 '24 10:06 Xia-zx

same problem

So-cean avatar Jan 09 '25 11:01 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

So-cean avatar Jan 09 '25 12:01 So-cean