Browse Source

!7358 fix nasnet script error

Merge pull request !7358 from panfengfeng/fix_nasnet_script_error
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
e0ccfa1328
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/nasnet/src/nasnet_a_mobile.py

+ 1
- 1
model_zoo/official/cv/nasnet/src/nasnet_a_mobile.py View File

@@ -911,7 +911,7 @@ class NASNetAMobileTrainOneStepWithClipGradient(nn.Cell):
self.network.add_flags(defer_inline=True)
self.weights = optimizer.parameters
self.optimizer = optimizer
self.grad = C.GradOperation('grad', get_by_list=True, sens_param=True)
self.grad = C.GradOperation(get_by_list=True, sens_param=True)
self.hyper_map = C.HyperMap()
self.sens = sens
self.reducer_flag = False


Loading…
Cancel
Save