Internal change

Mohamed Hammad · recml authors · commit 693ee8582882 · 2025-04-28T10:17:04.000-07:00
PiperOrigin-RevId: 750789052
diff --git a/recml/core/utils/keras_utils.py b/recml/core/utils/keras_utils.py
@@ -77,6 +77,7 @@ def save_model_variables(
         metrics_variables=False,
     )
     logging.info("Writing checkpoint for epoch %s...", epoch)
+
     self.save(step=epoch, items=state, metrics=logs)
 
   def restore_model_variables(self, model: keras.Model, epoch: int):

Original file line number	Diff line number	Diff line change
`@@ -77,6 +77,7 @@ def save_model_variables(`
`77`	`77`	`metrics_variables=False,`
`78`	`78`	`)`
`79`	`79`	`logging.info("Writing checkpoint for epoch %s...", epoch)`
	`80`	`+`
`80`	`81`	`self.save(step=epoch, items=state, metrics=logs)`
`81`	`82`
`82`	`83`	`def restore_model_variables(self, model: keras.Model, epoch: int):`