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

Fix sanity errors.

Browse Source
This commit is contained in:
Abdullah Selek 2019-07-04 09:09:33 +01:00
parent fe5d8a5cad
commit 8cc20365f7
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/training/saver.py
View File

@ -1276,11 +1276,12 @@ class Saver(object):
if save_path is None: if save_path is None:
raise ValueError("Can't load save_path when it is None.") raise ValueError("Can't load save_path when it is None.")
if not checkpoint_management.checkpoint_exists_internal(compat.as_text(save_path)): checkpoint_prefix = compat.as_text(save_path)
if not checkpoint_management.checkpoint_exists_internal(checkpoint_prefix):
raise ValueError("The passed save_path is not a valid checkpoint: " + raise ValueError("The passed save_path is not a valid checkpoint: " +
compat.as_text(save_path)) checkpoint_prefix)
logging.info("Restoring parameters from %s", compat.as_text(save_path)) logging.info("Restoring parameters from %s", checkpoint_prefix)
try: try:
if context.executing_eagerly(): if context.executing_eagerly():
self._build_eager(save_path, build_save=False, build_restore=True) self._build_eager(save_path, build_save=False, build_restore=True)