Skip to content

GitLab

Commit 60cbe4e4 authored by ayushmankumar7's avatar ayushmankumar7
Browse files

compat.v1.name_scope to tf.name_scope

parent 965cc3ee
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/image_classification/resnet/common.py
View file @ 60cbe4e4
...@@ -76,7 +76,7 @@ class PiecewiseConstantDecayWithWarmup( ...@@ -76,7 +76,7 @@ class PiecewiseConstantDecayWithWarmup(
def _get_learning_rate(self, step): def _get_learning_rate(self, step):
"""Compute learning rate at given 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.rescaled_lr, self.step_boundaries,
self.lr_values, self.warmup_steps, self.lr_values, self.warmup_steps,
self.compute_lr_on_cpu]): self.compute_lr_on_cpu]):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment