Internal change.

PiperOrigin-RevId: 658008404
google · Aug 1, 2024 · 80df97c · 80df97c
1 parent 53250b3
commit 80df97c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/checkpoint/orbax/checkpoint/checkpoint_manager.py b/checkpoint/orbax/checkpoint/checkpoint_manager.py
@@ -869,10 +869,12 @@ def should_save(self, step: int) -> bool:
       return False
     # If present then prefer should_save_fn over other 'save_*' options.
     if self._options.should_save_fn is not None:
-      logging.info(
+      logging.log_every_n(
+          logging.INFO,
           'CheckpointManagerOptions.should_save_fn is available, following save'
           ' options will be ignored: save_interval_steps=%s and'
           ' save_on_steps=%s',
+          500,
           self._options.save_interval_steps,
           self._options.save_on_steps,
       )