From 1bd2f100be71428a03727e6718e8d186551c9270 Mon Sep 17 00:00:00 2001 From: buxue Date: Mon, 7 Dec 2020 14:38:12 +0800 Subject: [PATCH] modify example of ParameterUpdate --- mindspore/nn/wrap/cell_wrapper.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/nn/wrap/cell_wrapper.py b/mindspore/nn/wrap/cell_wrapper.py index c330bea00d..544dbacd3e 100644 --- a/mindspore/nn/wrap/cell_wrapper.py +++ b/mindspore/nn/wrap/cell_wrapper.py @@ -403,12 +403,12 @@ class ParameterUpdate(Cell): ``Ascend`` Examples: - >>> network = Net() - >>> param = network.parameters_dict()['learning_rate'] + >>> network = nn.Dense(3, 4) + >>> param = network.parameters_dict()['weight'] >>> update = nn.ParameterUpdate(param) >>> update.phase = "update_param" - >>> lr = Tensor(0.001, mindspore.float32) - >>> update(lr) + >>> weight = Tensor(0.001, mindspore.float32) + >>> update(weight) """ def __init__(self, param):