のねのBlog

パソコンの問題や、ソフトウェアの開発で起きた問題など書いていきます。よろしくお願いします^^。

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/"