docs

Author: rohitgr7

pytorch_lightning/core/hooks.py

@@ -525,7 +525,7 @@ def transfer_batch_to_device(self, batch: Any, device: Optional[torch.device] =
 
         Example::
 
-            def transfer_batch_to_device(self, batch, device)
+            def transfer_batch_to_device(self, batch, device):
                 if isinstance(batch, CustomBatch):
                     # move all tensors in your custom data structure to the device
                     batch.samples = batch.samples.to(device)