github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-04 19:16:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

update doc

Browse Source
This commit is contained in:
oahzxl 2023-01-10 10:44:07 +08:00
parent 7d4abaa525
commit 615e7e68d9
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/autochunk/search_chunk.py
View File

@ -88,7 +88,7 @@ class SearchChunk(object):
Args: Args:
active_node (List): active node status for every node active_node (List): active node status for every node
peak_node (Node): peak memory node peak_node (Node): peak memory node
chunk_regions (List): chunk region info chunk_regions (List): chunk region infos
Returns: Returns:
chunk_region_start (int) chunk_region_start (int)