From 08a6a250c7d95e45ed1593d5a7f6eea679347a57 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 7 Aug 2019 13:17:51 -0400 Subject: [PATCH] updated docs --- docs/Trainer/Logging.md | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/docs/Trainer/Logging.md b/docs/Trainer/Logging.md index 378a8ffa..1b6174b9 100644 --- a/docs/Trainer/Logging.md +++ b/docs/Trainer/Logging.md @@ -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.