tensorflowはpbフォーマットモデルとして保存されます
2770 ワード
モデルの保存
if i % 200 == 0:
print("After %d training step(s), loss on training batch is %g." % (step, loss_value))
saver.save(sess, os.path.join(MODEL_SAVE_PATH, MODEL_NAME), global_step=global_step)
# ckpt , pb ,
constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ["output"])#["output"] , operation
#convert_variables_to_constants() , , , 。
with tf.gfile.FastGFile('model.pb', mode='wb') as f:
f.write(constant_graph.SerializeToString())# save/model.pb ,