Merge pull request #8436 from ayushmankumar7:name

PiperOrigin-RevId: 308469410

official/vision/image_classification/resnet/common.py

@@ -76,10 +76,7 @@ class PiecewiseConstantDecayWithWarmup(
 
   def _get_learning_rate(self, step):
     """Compute learning rate at given step."""
-    with tf.compat.v1.name_scope(self.name, 'PiecewiseConstantDecayWithWarmup',
-                                 [self.rescaled_lr, self.step_boundaries,
-                                  self.lr_values, self.warmup_steps,
-                                  self.compute_lr_on_cpu]):
+    with tf.name_scope('PiecewiseConstantDecayWithWarmup'):
       def warmup_lr(step):
         return self.rescaled_lr * (
             tf.cast(step, tf.float32) / tf.cast(self.warmup_steps, tf.float32))