From 12ee3c60dd4f9725f0206ccfb25fd3a266d5cfac Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 8 Jul 2019 20:12:27 -0400 Subject: [PATCH] docs --- pytorch_lightning/models/trainer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 597aa540..1d966188 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -1,3 +1,6 @@ +""" +The trainer handles all the logic for running a val loop, training loop, distributing, etc... +""" from time import sleep import subprocess import traceback @@ -23,6 +26,7 @@ try: except ModuleNotFoundError: APEX_AVAILABLE = False + class Trainer(TrainerIO): def __init__(self,