diff --git a/mindspore/train/callback.py b/mindspore/train/callback.py index 50fe5fcb26..e691cfd837 100644 --- a/mindspore/train/callback.py +++ b/mindspore/train/callback.py @@ -143,6 +143,11 @@ class CheckpointConfig: """ The config for model checkpoint. + Note: + During the training process, if dataset is transmitted through the data channel, + suggest set save_checkpoint_steps be an integer multiple of loop_size. + Otherwise there may be deviation in the timing of saving checkpoint. + Args: save_checkpoint_steps (int): Steps to save checkpoint. Default: 1. save_checkpoint_seconds (int): Seconds to save checkpoint. Default: 0. @@ -512,7 +517,7 @@ class ModelCheckpoint(Callback): Args: prefix (str): Checkpoint files names prefix. Default: "CKP". - directory (str): Lolder path into which checkpoint files will be saved. Default: None. + directory (str): Folder path into which checkpoint files will be saved. Default: None. config (CheckpointConfig): Checkpoint strategy config. Default: None. Raises: diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index 696b140462..4e6e67e32b 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -232,6 +232,7 @@ def load_param_into_net(net, parameter_dict): raise TypeError(msg) logger.info("Execute load parameter into net process.") + net.init_parameters_data() param_not_load = [] for _, param in net.parameters_and_names(): if param.name in parameter_dict: