github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-22 13:41:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed apex import (#227)

Browse Source
This commit is contained in:
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