Skip to content

GitLab

Commit 615e7e68 authored by oahzxl's avatar oahzxl
Browse files

update doc

parent 7d4abaa5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/autochunk/search_chunk.py
View file @ 615e7e68
...@@ -88,7 +88,7 @@ class SearchChunk(object): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment