From 615e7e68d9bc00e1b29879e73df4cca8afdd907d Mon Sep 17 00:00:00 2001 From: oahzxl Date: Tue, 10 Jan 2023 10:44:07 +0800 Subject: [PATCH] update doc --- colossalai/autochunk/search_chunk.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)