mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
added direct links to docs
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user