experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Stop Model.save_weights from printing a deprication warning (TF format)

Browse Source 
Also has it save relative paths in the checkpoint proto, which removes a behavior difference between it and CheckpointManager.

PiperOrigin-RevId: 227586345
This commit is contained in:
Allen Lavoie 2019-01-02 14:53:55 -08:00 committed by TensorFlower Gardener
parent 85ca0e9b81
commit 790a635a04
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/keras/engine/network.py
View File

@ -1382,9 +1382,10 @@ class Network(base_layer.Layer):
% (optimizer,)) % (optimizer,))
self._checkpointable_saver.save(filepath, session=session) self._checkpointable_saver.save(filepath, session=session)
# Record this checkpoint so it's visible from tf.train.latest_checkpoint. # 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), save_dir=os.path.dirname(filepath),
model_checkpoint_path=filepath, model_checkpoint_path=filepath,
save_relative_paths=True,
all_model_checkpoint_paths=[filepath]) all_model_checkpoint_paths=[filepath])
def load_weights(self, filepath, by_name=False): def load_weights(self, filepath, by_name=False):