При создании модели в test.py вызывается функция load_network из base_model.py. В этой функции происходит загрузка модели с помощью команды load_net = torch.load(load_path)[param_key]. Однако возникает ошибка, связанная с ключом param_key.

Вот информация об ошибке:
2020-09-27 11:17:56,991 INFO: Loading RRDBNet model from E:/Project/SRFramework/basicsr/models/esrgan/ESRGAN_x4.pth.
Traceback (most recent call last):
File "test.py", line 74, in
main()
File "test.py", line 61, in main
model = create_model(opt)
File "E:\Project\SRFramework\basicsr\models_init_.py", line 39, in create_model
model = model_cls(opt)
File "E:\Project\SRFramework\basicsr\models\sr_model.py", line 32, in init
self.opt['path']['strict_load'])
File "E:\Project\SRFramework\basicsr\models\base_model.py", line 261, in load_network
load_net = torch.load(load_path)[param_key]
KeyError: 'params'