From fa1c09bf5e95b83898417222771ed88303a5689e Mon Sep 17 00:00:00 2001 From: Juanyong Duan Date: Thu, 30 Jan 2025 21:43:06 +0800 Subject: [PATCH] update torch load method --- src/anomaly_detector/multivariate/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/anomaly_detector/multivariate/model.py b/src/anomaly_detector/multivariate/model.py index d43d785..2f1609c 100644 --- a/src/anomaly_detector/multivariate/model.py +++ b/src/anomaly_detector/multivariate/model.py @@ -151,7 +151,7 @@ def save_checkpoint(self): self.model.to(self.config.device) def load_checkpoint(self, model_path): - ckpt = torch.load(model_path) + ckpt = torch.load(model_path, weights_only=True) self.config = ckpt["config"] self.model = MultivariateGraphAttnDetector(self.config) self.model.load_state_dict(ckpt["state_dict"])