Skip to content

GitLab

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

Internal change 

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