adding tests

This commit is contained in:
William Falcon
2019-07-24 07:19:50 -04:00
parent 0527a4214b
commit 1eda58fa93
11 changed files with 1 additions and 2 deletions
@@ -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](../../examples/new_project_templates/lightning_module_template.py) and modify accordingly.
The easiest thing to do is copy [this template](../../pytorch_lightning/examples/new_project_templates/lightning_module_template.py) and modify accordingly.
Otherwise, to Define a Lightning Module, implement the following methods: