def serving_input_fn():
label_ids = tf.placeholder(tf.int32, [None, 3], name=’label_ids’)
input_ids = tf.placeholder(tf.int32, [None, 200], name=’input_ids’)
input_mask = tf.placeholder(tf.int32, [None, 200], name=’input_mask’)
segment_ids = tf.placeholder(tf.int32, [None, 200], name=’segment_ids’)
input_fn = tf.estimator.export.build_raw_serving_input_receiver_fn({
‘label_ids’: label_ids,
‘input_ids’: input_ids,
‘input_mask’: input_mask,
‘segment_ids’: segment_ids,
})()
return input_fn
在eval阶段生成pb格式，eval时加入如下代码：
if FLAGS.do_eval:
estimator._export_to_tpu = False
estimator.export_savedmodel(“pbmodels”, serving_input_fn)
Original: https://blog.csdn.net/sinat_34080511/article/details/121455781
Author: sinat_34080511
Title: tensorflow模型转化为pb格式
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/514098/
转载文章受原作者版权保护。转载请注明原作者出处！