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

debugging and gpu guide

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