diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index 41c69251..98c2b99b 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -27,4 +27,4 @@ This is the basic use of the trainer: from .trainer import Trainer -__all__ = ['Trainer'] \ No newline at end of file +__all__ = ['Trainer'] diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index a9e0ffb4..315d9ca9 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -1,6 +1,7 @@ import os import sys import warnings +import logging import torch import torch.distributed as dist @@ -46,7 +47,9 @@ class Trainer(TrainerIOMixin, TrainerTrainLoopMixin, TrainerCallbackConfigMixin, ): - r"""Abstract base class used to build new callbacks. + r""" + Main trainer class + """ def __init__( @@ -417,6 +420,7 @@ class Trainer(TrainerIOMixin, """ # # .. warning:: Following arguments become deprecated and they will be removed in v0.8.0: + # - `nb_sanity_val_steps` # Transfer params