From 196700a17f9e4c331c514ccb0eaf4601655f850b Mon Sep 17 00:00:00 2001 From: Sou Uchida Date: Fri, 6 Mar 2020 00:34:50 +0000 Subject: [PATCH] fix docs --- pytorch_lightning/loggers/trains.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pytorch_lightning/loggers/trains.py b/pytorch_lightning/loggers/trains.py index e67b1342d2ce7..48130f7b2aad2 100644 --- a/pytorch_lightning/loggers/trains.py +++ b/pytorch_lightning/loggers/trains.py @@ -76,9 +76,9 @@ def __init__( def experiment(self) -> trains.Task: r"""Actual TRAINS object. To use TRAINS features do the following. - Example:: - - self.logger.experiment.some_trains_function() + Example: + .. code-block:: python + self.logger.experiment.some_trains_function() """ return self._trains @@ -184,7 +184,7 @@ def log_image( Debug image to log. Can be one of the following types: Torch, Numpy, PIL image, path to image file (str) If Numpy or Torch, the image is assume to be the following: - shape: CHW or NCHW + shape: CHW color space: RGB value range: [0., 1.] (float) or [0, 255] (uint8) step (Optional[int], optional):