infer model load error
def restore_model(self, saver, model_dir): ckpt = tf.train.get_checkpoint_state(model_dir) if ckpt: saver.restore(self.sess, ckpt.model_checkpoint_path) print("restored model %s" % model_dir) else: print("fail to restore model %s" % model_dir) <==こっち、でエラーに、なっている。
tf.train.get_checkpoint_state
tf.train.get_checkpoint_state | TensorFlow Core r2.0
fail to restore model /content/log_train/sub_20190225_162939/checkpoint/experiment_0_batch_16/unet.model-5700.index
フォルダの位置にしたら、読み込むようになった。
MODEL_DIR = "/content/" + LOG_DIR_SUB + "/checkpoint/experiment_0_batch_16/" /content/log_train/sub_20190225_162939/checkpoint/experiment_0_batch_16/"