def get_checkpoint_state(self):
if self.ser_path.is_dir():
return tf.train.get_checkpoint_state(self.ser_path)
else:
return tf.train.get_checkpoint_state(self.ser_path.parent)
@check_path_exists()
After Change
if self.load_path.parent.is_dir():
return tf.train.get_checkpoint_state(self.load_path.parent)
else:
warn("Provided `load_path` is incorrect!")
else:
warn("No `load_path` is provided for {}".format(self.__class__.__name__)")
@overrides