MrGemy95/Tensorflow-Project-Template

About BaseModel.load

ulamaca opened this issue · 1 comments

To load the model properly, I think we need:
def load(self, sess):
latest_checkpoint = tf.train.latest_checkpoint(self.config.checkpoint_dir)
...

Instead of the original
latest_checkpoint = tf.train.latest_checkpoint(os.path.join(self.config.checkpoint_dir, self.config.exp_name))

yes, you're right
i modified it.
Thank you.