From f728fa05b5bbb71009bd964d835fa29a6f2adafc Mon Sep 17 00:00:00 2001 From: Anna Hulita Date: Thu, 24 Oct 2024 11:08:07 +0300 Subject: [PATCH] Update base_model.py added weights_only=False to torch.load() --- basicsr/models/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/basicsr/models/base_model.py b/basicsr/models/base_model.py index fbf8229f5..a3c8ac4c2 100644 --- a/basicsr/models/base_model.py +++ b/basicsr/models/base_model.py @@ -299,7 +299,7 @@ def load_network(self, net, load_path, strict=True, param_key='params'): """ logger = get_root_logger() net = self.get_bare_model(net) - load_net = torch.load(load_path, map_location=lambda storage, loc: storage) + load_net = torch.load(load_path, map_location=lambda storage, loc: storage, weights_only=False) if param_key is not None: if param_key not in load_net and 'params' in load_net: param_key = 'params'