Skip to content

S
seminar-breakout
Commit 0e69750a authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

ignore saver error

parent 2edcf101
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/callbacks/saver.py
View file @ 0e69750a
...@@ -81,7 +81,8 @@ class ModelSaver(Callback): ...@@ -81,7 +81,8 @@ class ModelSaver(Callback):
write_meta_graph=False) write_meta_graph=False)
logger.info("Model saved to %s." % tf.train.get_checkpoint_state(self.checkpoint_dir).model_checkpoint_path) logger.info("Model saved to %s." % tf.train.get_checkpoint_state(self.checkpoint_dir).model_checkpoint_path)
except (IOError, tf.errors.PermissionDeniedError, except (IOError, tf.errors.PermissionDeniedError,
tf.errors.ResourceExhaustedError): # disk error sometimes.. just ignore it tf.errors.ResourceExhaustedError,
tf.errors.AlreadyExistsError): # disk error sometimes.. just ignore it
logger.exception("Exception in ModelSaver!") logger.exception("Exception in ModelSaver!")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment