Merge pull request #7 from ashutosh1919/comments

Replaced Depricated tf.contrib.deprecated.scalar_summary with tf.compat.v1.summary.scalar
google · Jul 27, 2019 · 76d6be2 · 76d6be2
2 parents e7c8eb1 + 8827968
commit 76d6be2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/code/cifar_train_baseline.py b/code/cifar_train_baseline.py
@@ -247,7 +247,8 @@ def train_inception_baseline(max_step_run):
           labels=one_hot_labels, logits=logits)
       total_loss = tf.reduce_mean(total_loss)
 
-      tf.contrib.deprecated.scalar_summary('Total Loss', total_loss)
+      # Using latest tensorflow ProtoBuf.
+      tf.compat.v1.summary.scalar('Total Loss', total_loss)
 
       decay_steps = int(
           num_samples_per_epoch / FLAGS.batch_size * FLAGS.num_epochs_per_decay)

diff --git a/code/resnet_model.py b/code/resnet_model.py
@@ -124,7 +124,7 @@ def build_model(self):
   def _build_train_op(self):
     """Build training specific ops for the graph."""
     self.lrn_rate = tf.constant(self.hps.lrn_rate, tf.float32)
-    tf.contrib.deprecated.scalar_summary('learning rate', self.lrn_rate)
+    tf.compat.v1.summary.scalar('learning rate', self.lrn_rate)
 
     trainable_variables = tf.trainable_variables()
     grads = tf.gradients(self.cost, trainable_variables)