From 054ac94bd1da2c20b0f5b5ee2c3ec177469c3908 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Thu, 13 Aug 2020 11:51:37 -0400
Subject: [PATCH] track batch size (#2950)

---
 docs/source/datamodules.rst | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/docs/source/datamodules.rst b/docs/source/datamodules.rst
index e61d7ffc0a..7a99f0d3cb 100644
--- a/docs/source/datamodules.rst
+++ b/docs/source/datamodules.rst
@@ -2,6 +2,13 @@ LightningDataModule
 ===================
 A datamodule is a shareable, reusable class that encapsulates all the steps needed to process data:
 
+.. figure:: https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/datamodules.gif
+    :width: 800
+
+|
+
+A datamodule encapsulates the five steps involved in data processing in PyTorch:
+
 1. Download / tokenize / process.
 2. Clean and (maybe) save to disk.
 3. Load inside :class:`~torch.utils.data.Dataset`.