update doc

2023-01-10 10:44:07 +08:00 · 2023-01-10 10:44:07 +08:00 · 615e7e68d9
parent 7d4abaa525
commit 615e7e68d9
1 changed files with 1 additions and 1 deletions
--- 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)