modify callback comment

This commit is contained in:
changzherui 2021-07-08 23:59:12 +08:00
parent 6627e20ba1
commit e2a46ab0fb
2 changed files with 5 additions and 5 deletions

View File

@ -81,7 +81,7 @@ class Callback:
Examples:
>>> from mindspore import Model, nn
>>> from mindspore.train.callback._callback import Callback
>>> from mindspore.train.callback import Callback
>>> class Print_info(Callback):
>>> def step_end(self, run_context):
>>> cb_params = run_context.original_args()

View File

@ -315,9 +315,9 @@ def load(file_name, **kwargs):
kwargs (dict): Configuration options dictionary.
- dec_key: Byte type key used for decryption. Tha valid length is 16, 24, or 32.
- dec_mode: Specifies the decryption mode, take effect when dec_key is set. Option: 'AES-GCM' | 'AES-CBC'.
Default: 'AES-GCM'.
- dec_key (bytes): Byte type key used for decryption. Tha valid length is 16, 24, or 32.
- dec_mode (str): Specifies the decryption mode, take effect when dec_key is set.
Option: 'AES-GCM' | 'AES-CBC'. Default: 'AES-GCM'.
Returns:
Object, a compiled graph that can executed by `GraphCell`.
@ -395,7 +395,7 @@ def load_checkpoint(ckpt_file_name, net=None, strict_load=False, filter_prefix=N
>>>
>>> ckpt_file_name = "./checkpoint/LeNet5-1_32.ckpt"
>>> param_dict = load_checkpoint(ckpt_file_name, filter_prefix="conv1")
>>> print(param_dict["conv2.weight]")
>>> print(param_dict["conv2.weight"])
Parameter (name=conv2.weight, shape=(16, 6, 5, 5), dtype=Float32, requires_grad=True
"""
ckpt_file_name, filter_prefix = _check_checkpoint_param(ckpt_file_name, filter_prefix)