From f9285787a18cdaf7e59671a5a55d2de6993b9fd1 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 15 Jan 2020 21:17:16 -0500 Subject: [PATCH] fixed left menu --- docs/source/index.rst | 3 ++- pytorch_lightning/core/lightning.py | 5 +++++ pytorch_lightning/trainer/trainer.py | 5 +++++ 3 files changed, 12 insertions(+), 1 deletion(-) diff --git a/docs/source/index.rst b/docs/source/index.rst index 2fde388c..db1a051c 100644 --- a/docs/source/index.rst +++ b/docs/source/index.rst @@ -19,7 +19,8 @@ PyTorch-Lightning Documentation :name: docs :caption: Python API - documentation + trainer + lightning-module .. toctree:: :maxdepth: 1 diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 3fca3968..d22b6ed1 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -1,3 +1,8 @@ +""" +LightningModule +==================================== + +""" import os import warnings import collections diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index a1133004..8a768abd 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -1,4 +1,7 @@ """ +AAA +==================================== + The trainer handles all the logic for running a val loop, training loop, distributing, etc.. . """ @@ -50,6 +53,8 @@ class Trainer(TrainerIOMixin, TrainerTrainLoopMixin, TrainerCallbackConfigMixin, ): + r"""Abstract base class used to build new callbacks. + """ def __init__( self,