compat.v1.name_scope to tf.name_scope

60cbe4e4 · ayushmankumar7 · 965cc3ee · 60cbe4e4
Commit 60cbe4e4 authored Apr 25, 2020 by ayushmankumar7
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/image_classification/resnet/common.py official/vision/image_classification/resnet/common.py +1 -1

No files found.
--- a/official/vision/image_classification/resnet/common.py
+++ b/official/vision/image_classification/resnet/common.py
@@ -76,7 +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',
+    with tf.name_scope(self.name, 'PiecewiseConstantDecayWithWarmup',
                                 [self.rescaled_lr, self.step_boundaries,
                                  self.lr_values, self.warmup_steps,
                                  self.compute_lr_on_cpu]):