From 0c38bd6fc250b1c66b0cbb696b09ca73f88a6475 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 16:23:33 -0500 Subject: [PATCH] cleared spaces --- pytorch_lightning/core/lightning.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 8c57cbf7..94ce9384 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -802,9 +802,6 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks): batch (torch.nn.Tensor): Current batch split_size (int): How big the split is - .. note:: Called in the training loop after on_batch_start if `truncated_bptt_steps > 0`. - Each returned batch split is passed separately to training_step(...). - Return: list of batch splits. Each split will be passed to forward_step to enable truncated back propagation through time. The default implementation splits root level Tensors and @@ -831,6 +828,10 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks): splits.append(batch_split) return splits + + .. note:: Called in the training loop after on_batch_start if `truncated_bptt_steps > 0`. + Each returned batch split is passed separately to training_step(...). + """ time_dims = [len(x[0]) for x in batch if isinstance( x, torch.Tensor) or isinstance(x, collections.Sequence)]