在这里,无涯教程将重点介绍TensorFlow中的MetaGraph形成,这将帮助了解TensorFlow中的导出模块,MetaGraph包含基本信息,这是训练,执行判断或对先前训练过的图进行推理所必需的。
def export_meta_graph(filename=None, collection_list=None, as_text=False): """this code writes `MetaGraphDef` to save_path/filename. Arguments: filename: Optional meta_graph filename including the path. collection_list: List of string keys to collect. as_text: If `True`, writes the meta_graph as an ASCII proto. Returns: A `MetaGraphDef` proto. """
下面是其中一种典型的使用模式-
# 建立模型... with tf.Session() as sess: # 使用模型... # 将模型导出到 /tmp/my-model.meta。 meta_graph_def=tf.train.export_meta_graph(filename='/tmp/my-model.meta')
祝学习愉快!(内容编辑有误?请选中要编辑内容 -> 右键 -> 修改 -> 提交!)
Python实战 · 从0到1搭建直播视频平台 -〔Barry〕