Skip to content

Commit

Permalink
Merge pull request #447 from datamol-io/fix-strategy
Browse files Browse the repository at this point in the history
Fix `strategy` argument for trainer
  • Loading branch information
DomInvivo committed Aug 28, 2023
2 parents 95ece97 + 0170dcc commit e424f21
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion graphium/config/_loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit e424f21

Please sign in to comment.