added direct links to docs

This commit is contained in:
William Falcon committed 2020-01-15 21:43:04 -05:00
1 parent 2f12f21f34
commit 9bccb4ccdd
2 files changed
+9 -6

No files matched your search

+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.