Skip to content

GitLab

"server/vscode:/vscode.git/clone" did not exist on "737d5781e45fc2c297451773dea5ca1355b9a71d"
Commit 03545dbf authored by Le Hou's avatar Le Hou Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 352912327
parent 68726442
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/modeling/progressive/trainer.py
View file @ 03545dbf
......@@ -267,7 +267,8 @@ class ProgressiveTrainer(trainer_lib.Trainer):
step_interval = self.config.trainer.checkpoint_interval
else:
step_interval = self.config.trainer.export_checkpoint_interval
if global_step_np % step_interval != 0:
if global_step_np % step_interval != 0 and (
global_step_np < self._config.trainer.train_steps):
logging.info('Not exporting checkpoints in global step: %d.',
global_step_np)
return
......
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