github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 17:46:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] fix initialize bug with zero (#442)

Browse Source
This commit is contained in:
Jiarui Fang
2022-03-17 13:16:22 +08:00
committed by GitHub
parent 725a39f4bd
commit 496cbb0760
12 changed files with 87 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/utils/commons/memory.py
View File

@@ -1,4 +1,3 @@
import imp
import torch
from colossalai.utils import get_current_device