track batch size (Lightning-AI#2954)

ruotianluo · Aug 13, 2020 · 2c935d0 · 2c935d0
1 parent 054ac94
commit 2c935d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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
 
 |