-
Notifications
You must be signed in to change notification settings - Fork 186
Open
Description
This is not a issue,but a optimized save method.
- import dependency
from tensorflow.python.framework.graph_util import convert_variables_to_constants
- Convert variables to constants and save
output_graph_def = convert_variables_to_constants(sess, sess.graph_def, output_node_names=
['output/outputs'])
with tf.gfile.FastGFile('model/CTNModel.pb', mode='wb') as f:
f.write(output_graph_def.SerializeToString())
Metadata
Metadata
Assignees
Labels
No labels