Skip to content

GitLab

Commit 35979d3b authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 359809206
parent bc01957e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/nlp/train_ctl_continuous_finetune.py
View file @ 35979d3b
......@@ -39,8 +39,8 @@ def main(_):
params = train_utils.parse_configuration(FLAGS)
model_dir = FLAGS.model_dir
train_utils.serialize_config(params, model_dir)
continuous_finetune_lib.run_continuous_finetune(FLAGS.mode, params, model_dir,
FLAGS.pretrain_steps)
continuous_finetune_lib.run_continuous_finetune(
FLAGS.mode, params, model_dir, pretrain_steps=FLAGS.pretrain_steps)
train_utils.save_gin_config(FLAGS.mode, model_dir)
......
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