added direct links to docs

This commit is contained in:
William Falcon
2020-01-15 21:43:04 -05:00
parent 2f12f21f34
commit 9bccb4ccdd
2 changed files with 9 additions and 6 deletions
+9 -3
View File
@@ -1,8 +1,14 @@
"""
LightningModule
====================================
The LightningModule is the "system recipe." It groups the following in one file:
- computational system definition
- computations done on forward
- training loop
- validation loop
- testing loop
- train, val, test dataloaders
- optimizers
"""
"""
import os
import warnings
import collections
-3
View File
@@ -1,7 +1,4 @@
"""
Trainer
====================================
The lightning Trainer abstracts best practices for running a training, val, test routine.
It calls parts of your model when it wants to hand over full control and otherwise makes
training assumptions which are now standard practice in AI research.