Skip to content

GitLab

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

fixed apex import (#227)

parent eb3fda4c
Show 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
try:
import apex.amp as apex_amp
except ImportError:
raise ImportError('Cannot import apex.amp correctly.')
pass
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