From 1514810c3a5939379c1816c495cc9407a01d2865 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 13:21:25 -0400 Subject: [PATCH] updated docs --- docs/LightningModule/properties.md | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/docs/LightningModule/properties.md b/docs/LightningModule/properties.md index 060f2ec8f3..890d9f2007 100644 --- a/docs/LightningModule/properties.md +++ b/docs/LightningModule/properties.md @@ -38,3 +38,17 @@ self.trainer.current_epoch ... ``` +## Debugging +The LightningModule also offers these tricks to help debug. + +--- +#### example_input_array +In the LightningModule init, you can set a dummy tensor for this property +to get a print out of sizes coming into and out of every layer. +```python +def __init__(self): + # put the dimensions of the first input to your system + self.example_input_array = torch.rand(5, 28 * 28) +``` + +