diff --git a/colossalai/autochunk/search_chunk.py b/colossalai/autochunk/search_chunk.py index 590567045..c9e5e5172 100644 --- a/colossalai/autochunk/search_chunk.py +++ b/colossalai/autochunk/search_chunk.py @@ -88,7 +88,7 @@ class SearchChunk(object): Args: active_node (List): active node status for every node peak_node (Node): peak memory node - chunk_regions (List): chunk region info + chunk_regions (List): chunk region infos Returns: chunk_region_start (int)