首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

tensorflow学习笔记(五):变量保存与导入

如果你还在纠结如何保存tensorflow训练好的模型参数,用这个方法就对了 The Saver class adds ops to save and restore variables to and from...import tensorflow as tf """ 变量声明,运算声明 例:w = tf.get_variable(name="vari_name", shape=[], dtype=tf.float32...#使用saver提供的简便方法去调用 save op saver.save(sess, "save_path/file_name.ckpt") #file_name.ckpt如果不存在的话,会自动创建...with tf.Session() as sess: #这样你就可以使用restore的变量替换掉初始化的变量的值,而其它初始化的值不受影响 sess.run(init) if restor_from_checkpoint...file.ckpt") 参考资料 https://www.tensorflow.org/versions/r0.12/how_tos/variables/index.html#saving-and-restoring

63420
领券