From d6902ca5aeaf9f324760b3bb937c57dadc206ff6 Mon Sep 17 00:00:00 2001 From: changzherui Date: Fri, 23 Oct 2020 07:55:03 +0800 Subject: [PATCH] add ckpt code example --- mindspore/train/serialization.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index 33527dd1fe..afeea85161 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -341,6 +341,11 @@ def load_param_into_net(net, parameter_dict, strict_load=False): Raises: TypeError: Argument is not a Cell, or parameter_dict is not a Parameter dictionary. + + Examples: + >>> net = LeNet5() + >>> param_dict = load_checkpoint("LeNet5-2_1875.ckpt") + >>> load_param_into_net(net, param_dict) """ if not isinstance(net, nn.Cell): logger.error("Failed to combine the net and the parameters.")