Browse Source

!7900 fix comment problem

Merge pull request !7900 from caozhou/fix_comment
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
127f94b925
2 changed files with 4 additions and 3 deletions
  1. +2
    -1
      mindspore/train/callback/_checkpoint.py
  2. +2
    -2
      mindspore/train/serialization.py

+ 2
- 1
mindspore/train/callback/_checkpoint.py View File

@@ -90,7 +90,8 @@ class CheckpointConfig:
Default: True. Integrated save function is only supported in automatic parallel scene, not supported
in manual parallel.
async_save (bool): Whether asynchronous execution saves the checkpoint to a file. Default: False.
saved_network (Cell): Network to be saved in checkpoint file. Default: None.
saved_network (Cell): Network to be saved in checkpoint file. If the saved_network has no relation
with the network in training, the initial value of saved_network will be saved. Default: None.

Raises:
ValueError: If the input_param is None or 0.


+ 2
- 2
mindspore/train/serialization.py View File

@@ -232,8 +232,8 @@ def load_checkpoint(ckpt_file_name, net=None, strict_load=False, filter_prefix=N
net (Cell): Cell network. Default: None
strict_load (bool): Whether to strict load the parameter into net. If False, it will load parameter
in the param_dict into net with the same suffix. Default: False
filter_prefix (Union[str, list[str], tuple[str]]): Parameter with the filter prefix will not be loaded.
Default: None.
filter_prefix (Union[str, list[str], tuple[str]]): Parameters starting with the filter_prefix
will not be loaded. Default: None.

Returns:
Dict, key is parameter name, value is a Parameter.


Loading…
Cancel
Save