Unverified Commit 1f4936f1 authored by Madhura Pande's avatar Madhura Pande Committed by GitHub
Browse files

Tensorboard summary writing for few params

Using this code on GPU, was not logging "learning_rate" and "gradient norm" etc. We need to use simple "tf.summary.scalar" instead of contrib one to log these params. Tested on Tensorflow 1.14.
parent 88c864f7
...@@ -140,7 +140,7 @@ def model_fn(features, labels, mode, params): ...@@ -140,7 +140,7 @@ def model_fn(features, labels, mode, params):
def record_scalars(metric_dict): def record_scalars(metric_dict):
for key, value in metric_dict.items(): for key, value in metric_dict.items():
tf.contrib.summary.scalar(name=key, tensor=value) tf.summary.scalar(name=key, tensor=value)
def get_learning_rate(learning_rate, hidden_size, learning_rate_warmup_steps): def get_learning_rate(learning_rate, hidden_size, learning_rate_warmup_steps):
......
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