fixed apex import (#227)

hpcaitech · Feb 15, 2022 · f5ca88e · f5ca88e
1 parent eb3fda4
commit f5ca88e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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