From 592e087df17f0498ee5400d2d0e997fa6735dd21 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 15 Jan 2020 20:50:46 -0500 Subject: [PATCH] finished callbacks --- pytorch_lightning/trainer/__init__.py | 4 +++- pytorch_lightning/trainer/auto_mix_precision.py | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index 96254c26..e8b9dc05 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -1,5 +1,7 @@ """ -# Trainer +Trainer +==================================== + The lightning trainer abstracts best practices for running a training, val, test routine. It calls parts of your model when it wants to hand over full control and otherwise makes diff --git a/pytorch_lightning/trainer/auto_mix_precision.py b/pytorch_lightning/trainer/auto_mix_precision.py index b28193c0..2915f246 100644 --- a/pytorch_lightning/trainer/auto_mix_precision.py +++ b/pytorch_lightning/trainer/auto_mix_precision.py @@ -1,3 +1,4 @@ + from abc import ABC try: