tensorflow1.X读取checkpoint(加载训练好的模型及参数)的两种方式

在模型训练的保存路径下面会保存有多个模型(如下图),下面介绍两种读取方式
在这里插入图片描述
1,获取最新的保存的一个模型,比如上图的model-5200

# model_path是checkpoints的路径,下面方法会自动寻找最新保存的一个模型
ckpt_file = tf.train.latest_checkpoint(model_path)
saver.restore(sess,ckpt_file)

2,获取某个迭代次数的模型,比如要指定获取上图的model-5199

ckpt_file=os.path.join(model_path,'model-5199')
saver.restore(sess,ckpt_file)

Done!!!

你可能感兴趣的:(tensorflow,python,tensorflow,python)