优化修正文档检测中的部分语法

2024-08-20 14:27:51 +08:00
parent 299b762cad
commit 896d2aaf9b
3 changed files with 6 additions and 7 deletions
--- a/object_detection/core/nnet/py_factory.py
+++ b/object_detection/core/nnet/py_factory.py
@@ -119,7 +119,7 @@ class NetworkFactory(object):
    def load_pretrained_params(self, pretrained_model):
        print("loading from {}".format(pretrained_model))
        with open(pretrained_model, "rb") as f:
-            params = torch.load(f)
+            params = torch.load(f, weights_only=False)
            self.model.load_state_dict(params)

    def load_params(self, iteration):