wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

debugging and gpu guide

Browse Source
This commit is contained in:
William Falcon committed 2019-06-27 14:29:44 -04:00
1 parent 6ea3cc326f
commit 6c9797cf87
5 files changed
+292 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

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: