Skip to content

GitLab

".github/workflows/cuda/cu116-Linux-env.sh" did not exist on "69f32b7fc91efd1d2c26f3b759e87509775ba4f0"
Commit 4ab0b381 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

internal change 

PiperOrigin-RevId: 332084664
parent b139a84b
Show 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):
params["enable_metrics_in_training"] = flags_obj.enable_metrics_in_training
params["steps_between_evals"] = flags_obj.steps_between_evals
params["enable_checkpointing"] = flags_obj.enable_checkpointing
params["save_weights_only"] = flags_obj.save_weights_only
self.distribution_strategy = distribute_utils.get_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