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
Show 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):
FLAGS.param_set = 'tiny'
FLAGS.use_synthetic_data = True
FLAGS.steps_between_evals = 1
FLAGS.train_steps = 2
FLAGS.train_steps = 1
FLAGS.validation_steps = 1
FLAGS.batch_size = 8
FLAGS.batch_size = 4
FLAGS.max_length = 1
FLAGS.num_gpus = 1
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