From a3bd66167baf204276682ca7945bcac06ccab929 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 27 Jul 2019 13:41:38 -0400 Subject: [PATCH] Update trainer.py --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index ff11150b26..4b741c9f12 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -24,7 +24,7 @@ from pytorch_lightning.utils.debugging import MisconfigurationException try: from apex import amp APEX_AVAILABLE = True -except ModuleNotFoundError: # pragma: no cover +except Exception: APEX_AVAILABLE = False