Browse Source

!9157 remove is_grad

From: @lijiaqi0612
Reviewed-by: @sanjaychan,@zhunaipan
Signed-off-by: @sanjaychan
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
56655a5d2c
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/train/model.py

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

@@ -547,7 +547,7 @@ class Model:
Examples: Examples:
>>> dataset = create_custom_dataset() >>> dataset = create_custom_dataset()
>>> net = Net() >>> net = Net()
>>> loss = nn.SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True)
>>> loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True)
>>> loss_scale_manager = FixedLossScaleManager() >>> loss_scale_manager = FixedLossScaleManager()
>>> optim = Momentum(params=net.trainable_params(), learning_rate=0.1, momentum=0.9) >>> optim = Momentum(params=net.trainable_params(), learning_rate=0.1, momentum=0.9)
>>> model = Model(net, loss_fn=loss, optimizer=optim, metrics=None, loss_scale_manager=loss_scale_manager) >>> model = Model(net, loss_fn=loss, optimizer=optim, metrics=None, loss_scale_manager=loss_scale_manager)
@@ -668,7 +668,7 @@ class Model:
Examples: Examples:
>>> dataset = create_custom_dataset() >>> dataset = create_custom_dataset()
>>> net = Net() >>> net = Net()
>>> loss = nn.SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True)
>>> loss = nn.SoftmaxCrossEntropyWithLogits(sparse=True)
>>> model = Model(net, loss_fn=loss, optimizer=None, metrics={'acc'}) >>> model = Model(net, loss_fn=loss, optimizer=None, metrics={'acc'})
>>> model.eval(dataset) >>> model.eval(dataset)
""" """


Loading…
Cancel
Save