Tensorflow加载预训练模型(保存模型 ckpt文件)

import tensorflow as tf
 
sess = tf.Session()
# 先加载图和变量
saver = tf.train.import_meta_graph('my_test_model-1000.meta')
saver.restore(sess, tf.train.latest_checkpoint('./'))

ReferenceTensorflow加载预训练模型和保存模型(ckpt文件)以及迁移学习finetuning

发布了1636 篇原创文章 · 获赞 341 · 访问量 221万+

猜你喜欢

转载自blog.csdn.net/tony2278/article/details/103930472