Skip to content

GitLab

Commit 4ab0b381 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

internal change 

PiperOrigin-RevId: 332084664
parent b139a84b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/nlp/transformer/transformer_main.py
View file @ 4ab0b381
...@@ -159,6 +159,7 @@ class TransformerTask(object): ...@@ -159,6 +159,7 @@ class TransformerTask(object):
params["enable_metrics_in_training"] = flags_obj.enable_metrics_in_training params["enable_metrics_in_training"] = flags_obj.enable_metrics_in_training
params["steps_between_evals"] = flags_obj.steps_between_evals params["steps_between_evals"] = flags_obj.steps_between_evals
params["enable_checkpointing"] = flags_obj.enable_checkpointing params["enable_checkpointing"] = flags_obj.enable_checkpointing
params["save_weights_only"] = flags_obj.save_weights_only
self.distribution_strategy = distribute_utils.get_distribution_strategy( self.distribution_strategy = distribute_utils.get_distribution_strategy(
distribution_strategy=flags_obj.distribution_strategy, distribution_strategy=flags_obj.distribution_strategy,
......
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