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,