search

csguoh / MambaIR

[ECCV2024] An official pytorch implement of the paper "MambaIR: A simple baseline for image restoration with state-space model".
Apache License 2.0
495 stars 42 forks source link

KeyError: "No object named 'MambaIRModel' found in 'model' registry!" #53

Open wjkbigface opened 3 months ago

wjkbigface commented 3 months ago

Name MambaIRModel is not found, use name: MambaIRModel_basicsr! Traceback (most recent call last): File "/root/autodl-tmp/MambaIR-main/basicsr/train.py", line 225, in train_pipeline(root_path) File "/root/autodl-tmp/MambaIR-main/basicsr/train.py", line 131, in train_pipeline model = build_model(opt) File "/root/miniconda3/envs/mambair/lib/python3.10/site-packages/basicsr/models/init.py", line 26, in build_model model = MODEL_REGISTRY.get(opt['model_type'])(opt) File "/root/miniconda3/envs/mambair/lib/python3.10/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 'MambaIRModel' found in 'model' registry!"

liziang297 commented 3 months ago

excuse me，have you resolved the problem？

csguoh commented 3 months ago

Hi, you may refer to this issue. Sorry for the late reply.