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 (#2954)

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

4
pytorch_lightning/trainer/__init__.py
View File

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