diff --git a/colossalai/amp/apex_amp/apex_amp.py b/colossalai/amp/apex_amp/apex_amp.py index b43228410088..6d7196b33466 100644 --- a/colossalai/amp/apex_amp/apex_amp.py +++ b/colossalai/amp/apex_amp/apex_amp.py @@ -5,7 +5,7 @@ try: import apex.amp as apex_amp except ImportError: - raise ImportError('Cannot import apex.amp correctly.') + pass from torch import Tensor