Load checkpoints with weights_only=False in torch estimator

GaetanLepage · GaetanLepage · commit 6420e75cfbea · 2025-11-29T22:04:44.000Z
diff --git a/src/gluonts/torch/model/estimator.py b/src/gluonts/torch/model/estimator.py
@@ -228,7 +228,8 @@ def train_model(
                 f"Loading best model from {checkpoint.best_model_path}"
             )
             best_model = training_network.__class__.load_from_checkpoint(
-                checkpoint.best_model_path
+                checkpoint.best_model_path,
+                weights_only=False,
             )
         else:
             best_model = training_network

Original file line number	Diff line number	Diff line change
`@@ -228,7 +228,8 @@ def train_model(`
`228`	`228`	`f"Loading best model from {checkpoint.best_model_path}"`
`229`	`229`	`)`
`230`	`230`	`best_model = training_network.__class__.load_from_checkpoint(`
`231`		`- checkpoint.best_model_path`
	`231`	`+ checkpoint.best_model_path,`
	`232`	`+ weights_only=False,`
`232`	`233`	`)`
`233`	`234`	`else:`
`234`	`235`	`best_model = training_network`