github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fixed apex import (#227)

pull/232/head v0.0.2
Frank Lee 2022-02-14 18:04:57 +08:00
parent eb3fda4c28
commit f5ca88ec97
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/amp/apex_amp/apex_amp.py
View File

@ -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