Fix strategy argument for trainer

datamol-io · Aug 27, 2023 · 0170dcc · 0170dcc
1 parent 95ece97
commit 0170dcc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphium/config/_loader.py b/graphium/config/_loader.py
@@ -371,7 +371,7 @@ def load_trainer(
     cfg_trainer = deepcopy(config["trainer"])
 
     # Define the IPU plugin if required
-    strategy = config["trainer"]["trainer"].get("strategy", "auto")
+    strategy = cfg_trainer["trainer"].pop("strategy", "auto")
     if accelerator_type == "ipu":
         ipu_opts, ipu_inference_opts = _get_ipu_opts(config)