fix nasnet GradOperation

4 years ago · 5537ec8d92
parent fa37f8fde3
commit 5537ec8d92
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