Skip to content

GitLab

Commit b62439d7 authored by saberkun's avatar saberkun
Browse files

Merge pull request #7712 from madhurapande19:patch-1 

PiperOrigin-RevId: 276609206
parents 18b1714c 1f4936f1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/transformer/transformer_main.py
View file @ b62439d7
...@@ -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