From 24897506a076e64f7896552d8845f9dfad7dad7e 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 | 1 - pytorch_lightning/trainer/auto_mix_precision.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/__init__.py b/pytorch_lightning/trainer/__init__.py index 3721a762..232e9a1c 100644 --- a/pytorch_lightning/trainer/__init__.py +++ b/pytorch_lightning/trainer/__init__.py @@ -1,6 +1,5 @@ """ 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: