!7358 fix nasnet script error

Merge pull request !7358 from panfengfeng/fix_nasnet_script_error
4 years ago · e0ccfa1328
parent e52588285b 5537ec8d92
commit e0ccfa1328
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/nasnet/src/nasnet_a_mobile.py
+++ b/model_zoo/official/cv/nasnet/src/nasnet_a_mobile.py
@ -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