Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

track batch size (#2950)

This commit is contained in:
William Falcon 2020-08-13 11:51:37 -04:00 committed by GitHub
parent 4354690e55
commit 054ac94bd1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docs/source/datamodules.rst
View File

@ -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`.