Skip to content

GitLab

Commit cd11210c authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 410021811
parent 65a967e9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/transformer/transformer_main_test.py
View file @ cd11210c
...@@ -53,9 +53,9 @@ class TransformerTaskTest(tf.test.TestCase): ...@@ -53,9 +53,9 @@ class TransformerTaskTest(tf.test.TestCase):
FLAGS.param_set = 'tiny' FLAGS.param_set = 'tiny'
FLAGS.use_synthetic_data = True FLAGS.use_synthetic_data = True
FLAGS.steps_between_evals = 1 FLAGS.steps_between_evals = 1
FLAGS.train_steps = 2 FLAGS.train_steps = 1
FLAGS.validation_steps = 1 FLAGS.validation_steps = 1
FLAGS.batch_size = 8 FLAGS.batch_size = 4
FLAGS.max_length = 1 FLAGS.max_length = 1
FLAGS.num_gpus = 1 FLAGS.num_gpus = 1
FLAGS.distribution_strategy = 'off' FLAGS.distribution_strategy = 'off'
......
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