Skip to content

Commit 01b7293

Browse files
committed
remove redundant todo
1 parent 91f1387 commit 01b7293

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

pytorch_lightning/accelerators/accelerator.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -174,8 +174,6 @@ def batch_to_device(
174174
dataloader_idx: The index of the dataloader to which the batch belongs.
175175
"""
176176
model = self.lightning_module
177-
178-
# TODO: Add support to allow batch transfer to device in Lightning for DP mode.
179177
if model is not None and not isinstance(self.training_type_plugin, DataParallelPlugin):
180178
# no need to transfer batch to device in DP mode
181179
return model._apply_batch_transfer_handler(batch, device, dataloader_idx)

0 commit comments

Comments
 (0)