mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-11 12:31:23 +08:00
updated docs
This commit is contained in:
@@ -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)
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user