modify the event pickle file

2024-07-28 19:21:27 +08:00
parent f14faa323e
commit 0ccfd0151f
152 changed files with 57 additions and 27 deletions
--- a/models/experimental.py
+++ b/models/experimental.py
@ -76,7 +76,7 @@ def attempt_load(weights, device=None, inplace=True, fuse=True):

    model = Ensemble()
    for w in weights if isinstance(weights, list) else [weights]:
-        ckpt = torch.load(attempt_download(w), map_location=device)  # load
+        ckpt = torch.load(attempt_download(w), map_location=device, weights_only=False)  # load
        ckpt = (ckpt.get('ema') or ckpt['model']).to(device).float()  # FP32 model

        # Model compatibility updates