From 790a635a044c9b33103f1e02f57faf08c86eee36 Mon Sep 17 00:00:00 2001 From: Allen Lavoie Date: Wed, 2 Jan 2019 14:53:55 -0800 Subject: [PATCH] Stop Model.save_weights from printing a deprication warning (TF format) Also has it save relative paths in the checkpoint proto, which removes a behavior difference between it and CheckpointManager. PiperOrigin-RevId: 227586345 --- tensorflow/python/keras/engine/network.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/keras/engine/network.py b/tensorflow/python/keras/engine/network.py index 0837ce6780c..8e130aef40c 100644 --- a/tensorflow/python/keras/engine/network.py +++ b/tensorflow/python/keras/engine/network.py @@ -1382,9 +1382,10 @@ class Network(base_layer.Layer): % (optimizer,)) self._checkpointable_saver.save(filepath, session=session) # Record this checkpoint so it's visible from tf.train.latest_checkpoint. - checkpoint_management.update_checkpoint_state( + checkpoint_management.update_checkpoint_state_internal( save_dir=os.path.dirname(filepath), model_checkpoint_path=filepath, + save_relative_paths=True, all_model_checkpoint_paths=[filepath]) def load_weights(self, filepath, by_name=False):