fixed apex import (#227)

2022-02-14 18:04:57 +08:00 · 2022-02-14 18:04:57 +08:00 · f5ca88ec97
parent eb3fda4c28
commit f5ca88ec97
1 changed files with 1 additions and 1 deletions
--- a/colossalai/amp/apex_amp/apex_amp.py
+++ b/colossalai/amp/apex_amp/apex_amp.py
@ -5,7 +5,7 @@ import torch.nn as nn
 try:
    import apex.amp as apex_amp
 except ImportError:
-    raise ImportError('Cannot import apex.amp correctly.')
+    pass

 from torch import Tensor