wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

added lightning model docs

Browse Source
This commit is contained in:
William Falcon
2019-06-27 10:24:08 -04:00
parent 3b7c7c65e4
commit fa12098c5f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/Pytorch-Lightning/LightningModule.md
+3 -1
View File
@@ -3,7 +3,9 @@
A lightning module is a strict superclass of nn.Module, it provides a standard interface for the trainer to interact with the model.
To Define a Lightning Module, implement the following methods:
The easiest thing to do is copy [this template](lightning_module_template.py) and modify accordingly.
Otherwise, to Define a Lightning Module, implement the following methods:
**Required**: