From 0af6c26e30a749dc3bbc4d666f791127ec0e43a3 Mon Sep 17 00:00:00 2001 From: yao_yf Date: Fri, 17 Dec 2021 09:43:28 +0800 Subject: [PATCH] restore_group_info api fix --- mindspore/python/mindspore/train/serialization.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/python/mindspore/train/serialization.py b/mindspore/python/mindspore/train/serialization.py index 38df0fc766e..e8d9f5c6175 100644 --- a/mindspore/python/mindspore/train/serialization.py +++ b/mindspore/python/mindspore/train/serialization.py @@ -1299,9 +1299,9 @@ def _merge_param_with_strategy(sliced_data, parameter_name, strategy, is_even): def restore_group_info_list(group_info_file_name): """ Build rank list, the checkpoint of ranks in the rank list has the same contents with the local rank - that saves the group_info_file_name. - To save the group info file, please export GROUP_INFO_FILE environment variables like - "export GROUP_INFO_FILE=/data/group_info.pb". + who saves the group_info_file_name. To save the group info file, please export GROUP_INFO_FILE environment variables + like "export GROUP_INFO_FILE=/data/group_info.pb". + Args: group_info_file_name (str): Name of group information file.