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 c2c3ab0 commit c96dddaCopy full SHA for c96ddda
pytorch_lightning/trainer/connectors/accelerator_connector.py
@@ -484,7 +484,7 @@ def _is_slurm_managing_tasks(self):
484
return num_slurm_tasks == total_requested_devices
485
486
def _choose_strategy(self):
487
- if self._accelerator_flag == "ipu_strategy":
+ if self._accelerator_flag == "ipu":
488
self._strategy_flag = "ipu_strategy"
489
elif self._accelerator_flag == "tpu":
490
if self._parallel_devices and len(self._parallel_devices) > 1:
0 commit comments