From fa12098c5fdf6e743a0634dcd9e55950049a6f39 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 27 Jun 2019 10:24:08 -0400 Subject: [PATCH] added lightning model docs --- docs/Pytorch-Lightning/LightningModule.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docs/Pytorch-Lightning/LightningModule.md b/docs/Pytorch-Lightning/LightningModule.md index d57c6b9e..e0c80526 100644 --- a/docs/Pytorch-Lightning/LightningModule.md +++ b/docs/Pytorch-Lightning/LightningModule.md @@ -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**: