updated docs

This commit is contained in:
William Falcon
2019-08-07 13:17:51 -04:00
parent 48bc3465e9
commit 08a6a250c7
+9
View File
@@ -52,6 +52,15 @@ Trainer(experiment=exp)
---
### Tensorboard support
In the LightningModule you can access the experiment logger by doing:
```python
self.experiment
# add image
# Look at PyTorch SummaryWriter docs for what you can do.
self.experiment.add_image(...)
```
The experiment object is a strict subclass of PyTorch SummaryWriter. However, this class
also snapshots every detail about the experiment (data folder paths, code, hyperparams),
and allows you to visualize it using tensorboard.