From 2c935d048e69a2890889dea768ecceb0252cf321 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 13 Aug 2020 12:40:54 -0400 Subject: [PATCH] track batch size (#2954) --- pytorch_lightning/trainer/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index 2b535315ad..51b4d68bfa 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -10,8 +10,8 @@ Once you've organized your PyTorch code into a LightningModule, the Trainer automates everything else. -.. figure:: /_images/lightning_module/pt_trainer.png - :alt: Convert from PyTorch to Lightning +.. figure:: https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/trainer_v1.gif + :width: 800 |