Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit 74e419b

Browse files
committed
Merge branch 'raw' of https://github.com/SkyAndCloud/tensor2tensor into raw
2 parents f8c337d + 3c04f3a commit 74e419b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensor2tensor/bin/t2t_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def create_run_config(hp):
178178
save_ckpt_secs = FLAGS.save_checkpoints_secs or None
179179
if save_ckpt_secs:
180180
save_ckpt_steps = None
181-
assert FLAGS.output_dir
181+
assert FLAGS.output_dir or FLAGS.checkpoint_path
182182
return trainer_lib.create_run_config(
183183
model_dir=os.path.expanduser(FLAGS.output_dir),
184184
master=FLAGS.master,

0 commit comments

Comments
 (0)