fix rltv imports

Lightning-AI · Aug 6, 2019 · eacd93e · eacd93e
1 parent 50cca25
commit eacd93e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/models/__init__.py b/pytorch_lightning/models/__init__.py
diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
@@ -23,7 +23,7 @@
 try:
     from apex import amp
     APEX_AVAILABLE = True
-except Exception:
+except ImportError:
     APEX_AVAILABLE = False