"scripts/git@developer.sourcefind.cn:Wenxuan/LightX2V.git" did not exist on "708ea9d7e06ee09df35f66150e21abc5ab40af5f"
Commit 9616954e authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

Adds save_steps to fix squad

PiperOrigin-RevId: 275994710
parent 128326ca
...@@ -142,5 +142,5 @@ flags.DEFINE_integer( ...@@ -142,5 +142,5 @@ flags.DEFINE_integer(
default=16, default=16,
help="Size of the test batch across all hosts.") help="Size of the test batch across all hosts.")
flags.DEFINE_integer( flags.DEFINE_integer(
"save_steps", default=None, help="Number of steps for saving checkpoint.") "save_steps", default=1000, help="Number of steps for saving checkpoint.")
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
...@@ -184,7 +184,7 @@ def main(unused_argv): ...@@ -184,7 +184,7 @@ def main(unused_argv):
optimizer=optimizer, optimizer=optimizer,
learning_rate_fn=learning_rate_fn, learning_rate_fn=learning_rate_fn,
model_dir=FLAGS.model_dir, model_dir=FLAGS.model_dir,
save_steps=1000) save_steps=FLAGS.save_steps)
if __name__ == "__main__": if __name__ == "__main__":
......
...@@ -288,7 +288,8 @@ def main(unused_argv): ...@@ -288,7 +288,8 @@ def main(unused_argv):
steps_per_loop=steps_per_loop, steps_per_loop=steps_per_loop,
optimizer=optimizer, optimizer=optimizer,
learning_rate_fn=learning_rate_fn, learning_rate_fn=learning_rate_fn,
model_dir=FLAGS.model_dir) model_dir=FLAGS.model_dir,
save_steps=FLAGS.save_steps)
if __name__ == "__main__": if __name__ == "__main__":
......
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