TensorFlow
iauhsnaiq
这个作者很懒,什么都没留下…
展开
-
TensorFlow持久化:模型保存与加载
目录checkpoint保存加载saved_model保存加载frozen_graph保存加载 checkpoint 保存 saver = tf.train.Saver( var_list=None, reshape=False, sharded=False, max_to_keep=5, keep_checkpoint_every_n_hours=10000.0, name=None, restore_sequentially=False, saver_def=None, buil原创 2020-09-19 08:30:30 · 164 阅读 · 0 评论 -
TensorFlow 默认计算图、默认会话、计算机制
目录计算图 和 默认计算图会话 和 默认会话 计算图 和 默认计算图 tensorflow 可以定义多个计算图,分别在不同的计算图里定义不同的计算。 但是,我发现: 想在计算图里定义变量、常量、张量、op等节点的话,必须要先将该计算图设置为“默认计算图”。 # 如果去掉 as_default() 就会报错 with tf.Graph().as_default() as graph: ... 会话 和 默认会话 想执行计算图,有两种方法: 一种是用会话的 run 方法,另一种是用节点自己的 run 或 e原创 2020-09-04 23:30:09 · 581 阅读 · 0 评论 -
TensorFlow 将 checkpoint 冻结为 frozen_graph
import tensorflow as tf from tensorflow.python.tools.freeze_graph import freeze_graph def freeze(ckpt_dir, meta_file_path, output_node_names): # 导入结构、加载权重 saver = tf.train.import_meta_graph(meta_file_path) with tf.Session() as sess: save原创 2020-08-31 16:56:56 · 492 阅读 · 0 评论