We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 08a6a25 commit 1514810Copy full SHA for 1514810
docs/LightningModule/properties.md
@@ -38,3 +38,17 @@ self.trainer.current_epoch
38
...
39
```
40
41
+## Debugging
42
+The LightningModule also offers these tricks to help debug.
43
+
44
+---
45
+#### example_input_array
46
+In the LightningModule init, you can set a dummy tensor for this property
47
+to get a print out of sizes coming into and out of every layer.
48
+```python
49
+def __init__(self):
50
+ # put the dimensions of the first input to your system
51
+ self.example_input_array = torch.rand(5, 28 * 28)
52
+```
53
54
0 commit comments