tf.train.latest_checkpoint返回None的解决办法

使用tf.train.latest_checkpoint(model_dir)方法加载与训练模型的时候,model_dir参数是希望加载的模型所在路径,作用是生成这个model(.ckpt)文件的地址

data = tf.train.latest_checkpoint('./model')

但是打印出data显示为None
解决方法是不用tf.train.latest_checkpoint,直接将路径+.ckpt文件名拼接起来形成文件地址

model_dir='.\model'
model_name='vgg_16.ckpt'
pretrain_path = os.path.join(model_dir,model_name)

你可能感兴趣的:(技术问题)