fix docs

bmartinn · Mar 6, 2020 · 196700a · 196700a
1 parent c2fd85b
commit 196700a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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):