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

[NFC] polish initializer_data.py code style (#3287)

Browse Source
pull/3313/head
RichardoLuo 2 years ago committed by binmakeswell
parent 1bed38ef37
commit 1ce9d0c531
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File

3
colossalai/context/process_group_initializer/initializer_data.py
Unescape View File

@ -4,8 +4,9 @@
from torch import distributed as dist from torch import distributed as dist
from colossalai.registry import DIST_GROUP_INITIALIZER from colossalai.registry import DIST_GROUP_INITIALIZER
from .process_group_initializer import ProcessGroupInitializer
from ..parallel_mode import ParallelMode from ..parallel_mode import ParallelMode
from .process_group_initializer import ProcessGroupInitializer
@DIST_GROUP_INITIALIZER.register_module @DIST_GROUP_INITIALIZER.register_module

Write Preview
Loading…
Cancel
Save