Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
four4fish committed Feb 3, 2022
1 parent c2c3ab0 commit c96ddda
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -484,7 +484,7 @@ def _is_slurm_managing_tasks(self):
return num_slurm_tasks == total_requested_devices

def _choose_strategy(self):
if self._accelerator_flag == "ipu_strategy":
if self._accelerator_flag == "ipu":
self._strategy_flag = "ipu_strategy"
elif self._accelerator_flag == "tpu":
if self._parallel_devices and len(self._parallel_devices) > 1:
Expand Down

0 comments on commit c96ddda

Please sign in to comment.