模型保存
github_元宝
这个作者很懒,什么都没留下…
展开
-
tensorboard查看tf saved_model
# coding: utf-8from __future__ import print_functionimport tensorflow.compat.v1 as tftf.disable_v2_behavior()sess=tf.Session()tf.train.import_meta_graph("../model/shakespeare/model-200.meta")tf.summary.FileWriter("log",sess.graph)sess.close()原创 2021-04-07 14:32:05 · 227 阅读 · 0 评论 -
tf2.x函数签名
一般说法:input_signature的好处:1.可以限定函数的输入类型,以防止调用函数时调错,2.一个函数有了input_signature之后,在tensorflow里边才可以保存成savedmodel。在保存成savedmodel的过程中,需要使用get_concrete_function函数把一个tf.function标注的普通的python函数变成带有图定义的函数。使用经验:当一个函数被@tf.function 比较后,可以使用get_concrete_function 然后传入原函数原创 2021-08-25 11:28:15 · 360 阅读 · 0 评论