updated docs

This commit is contained in:
William Falcon 2019-08-07 13:21:25 -04:00
parent 08a6a250c7
commit 1514810c3a
1 changed files with 14 additions and 0 deletions

View File

@ -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)
```