diff --git a/pytorch_lightning/models/__init__.py b/pytorch_lightning/models/__init__.py new file mode 100644 index 0000000000000..e69de29bb2d1d diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 2655809c4a7de..2a73d8678b918 100644 --- 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