We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4d472d7 commit 81d1d9dCopy full SHA for 81d1d9d
graphium/config/_loader.py
@@ -372,7 +372,7 @@ def load_trainer(
372
cfg_trainer = deepcopy(config["trainer"])
373
374
# Define the IPU plugin if required
375
- strategy = "auto"
+ strategy = config["trainer"]["trainer"].get("strategy", "auto")
376
if accelerator_type == "ipu":
377
ipu_opts, ipu_inference_opts = _get_ipu_opts(config)
378
@@ -385,6 +385,9 @@ def load_trainer(
385
precision=config["trainer"]["trainer"].get("precision"),
386
)
387
388
+ if strategy != "auto":
389
+ raise ValueError("IPUs selected, but strategy is not set to 'auto'")
390
+
391
from lightning_graphcore import IPUStrategy
392
393
strategy = IPUStrategy(training_opts=training_opts, inference_opts=inference_opts)
0 commit comments