jixiaozhong / RealSR

Real-World Super-Resolution via Kernel Estimation and Noise Injection
Apache License 2.0
740 stars 102 forks source link

No module named 'RRDBNet_arch #40

Open Worrah opened 2 years ago

Worrah commented 2 years ago

hello. i've got this error when trying to run test.py what am i doing wrong?

22-09-05 05:02:28.757 - INFO: Loading model for G [D:\Bots\RealSR\model_new.pth] ... Traceback (most recent call last): File "D:\Bots\RealSR\test.py", line 35, in model = create_model(opt) File "D:\Bots\RealSR\models__init.py", line 18, in create_model m = M(opt) File "D:\Bots\RealSR\models\SRGAN_model.py", line 127, in init__ self.load() # load G and D if needed File "D:\Bots\RealSR\models\SRGAN_model.py", line 352, in load self.load_network(load_path_G, self.netG, self.opt['path']['strict_load']) File "D:\Bots\RealSR\models\base_model.py", line 90, in load_network load_net = torch.load(load_path) File "C:\Users\Wolh\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\serialization.py", line 713, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "C:\Users\Wolh\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\serialization.py", line 930, in _legacy_load result = unpickler.load() File "C:\Users\Wolh\AppData\Local\Programs\Python\Python310\lib\site-packages\torch\serialization.py", line 746, in find_class return super().find_class(mod_name, name) ModuleNotFoundError: No module named 'RRDBNet_arch'