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

update doc

pull/2364/head
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)