We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 66d5a04 commit af65796Copy full SHA for af65796
python/rapidocr/inference_engine/torch.py
@@ -53,7 +53,7 @@ def __init__(self, cfg) -> None:
53
54
arch_config = all_arch_config.get(file_name)
55
self.predictor = BaseModel(arch_config)
56
- self.predictor.load_state_dict(torch.load(model_path, weights_only=True))
+ self.predictor.load_state_dict(torch.load(model_path, weights_only=False))
57
self.predictor.eval()
58
59
self.use_gpu = False
0 commit comments