From e2a46ab0fbae23a9681d63722bbd1def1d5f3439 Mon Sep 17 00:00:00 2001 From: changzherui Date: Thu, 8 Jul 2021 23:59:12 +0800 Subject: [PATCH] modify callback comment --- mindspore/train/callback/_callback.py | 2 +- mindspore/train/serialization.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/mindspore/train/callback/_callback.py b/mindspore/train/callback/_callback.py index a57c57c9494..0b30feeae7c 100644 --- a/mindspore/train/callback/_callback.py +++ b/mindspore/train/callback/_callback.py @@ -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() diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index 572839fd60f..7f21126ef60 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -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)