Skip to content

Commit ed83e26

Browse files
committed
change num_devices to use device_ids
1 parent f974bc6 commit ed83e26

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch_lightning/trainer/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2025,7 +2025,7 @@ def device_ids(self) -> List[int]:
20252025
@property
20262026
def num_devices(self) -> int:
20272027
"""Number of devices the trainer uses per node."""
2028-
return len(self.strategy.parallel_devices) if isinstance(self.strategy, ParallelStrategy) else 1
2028+
return len(self.device_ids)
20292029

20302030
@property
20312031
def num_processes(self) -> int:
@@ -2048,7 +2048,7 @@ def num_gpus(self) -> int:
20482048
return self._accelerator_connector.num_gpus
20492049

20502050
@property
2051-
def devices(self) -> Optional[Union[List[int], str, int]]:
2051+
def devices(self) -> int:
20522052
rank_zero_deprecation(
20532053
"`Trainer.devices` was deprecated in v1.6 and will be removed in v1.8."
20542054
" Please use `Trainer.num_devices` or `Trainer.device_ids` to get device information instead."

0 commit comments

Comments
 (0)