added direct links to docs

2026-09-09 11:32:07 +08:00 · 2020-01-16 07:27:22 -05:00
parent 4b08974d4d
commit d34de3890e
1 changed files with 9 additions and 3 deletions
@@ -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