diff --git a/pytorch_lightning/models/__init__.py b/pytorch_lightning/models/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 2655809c..2a73d867 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -23,7 +23,7 @@ from ..utilities.debugging import MisconfigurationException try: from apex import amp APEX_AVAILABLE = True -except Exception: +except ImportError: APEX_AVAILABLE = False