Skip to content

GitLab

Commit f5ca88ec authored by Frank Lee's avatar Frank Lee
Browse files

fixed apex import (#227)

parent eb3fda4c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/amp/apex_amp/apex_amp.py
View file @ f5ca88ec
...@@ -5,7 +5,7 @@ import torch.nn as nn ...@@ -5,7 +5,7 @@ import torch.nn as nn
try: try:
import apex.amp as apex_amp import apex.amp as apex_amp
except ImportError: except ImportError:
raise ImportError('Cannot import apex.amp correctly.') pass
from torch import Tensor from torch import Tensor
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment