fix centernet loss error.

model_zoo/research/cv/centernet/src/centernet_pose.py

@@ -308,11 +308,8 @@ class CenterNetWithLossScaleCell(nn.Cell):
             cond = self.less_equal(self.base, flag_reduce)
         else:
             cond = self.less_equal(self.base, flag_sum)
-        overflow = cond
-        if overflow:
-            succ = False
-        else:
-            succ = self.optimizer(grads)
+
+        succ = self.optimizer(grads)
         ret = (loss, cond, scaling_sens)
         return ops.depend(ret, succ)
 

model_zoo/research/cv/centernet/src/utils.py

@@ -137,7 +137,7 @@ class GatherFeature(nn.Cell):
             self.gather_nd = ops.GatherD()
             self.expand_dims = ops.ExpandDims()
         else:
-            self.gather_nd = ops.GatherND()
+            self.gather_nd = ops.GatherNd()
 
     def construct(self, feat, ind):
         """gather by specified index"""