如何将Graph Transform Tool与通过Estimator API导出的保存模型一起使用

时间:2018-07-30 21:19:08

标签: tensorflow proto

如果使用高级估算器API对模型进行训练,则可以通过调用导出模型

estimator.export_savedmodel(export_dir_base, serving_input_receiver_fn,
                            strip_default_attrs=True)

但是,通过这种方法,将使用以下目录结构保存模型:

variables/
    variables.data-?????-of-?????
    variables.index
saved_model.pb|saved_model.pbtxt 

我找不到将这种样式(称为savemodel格式)转换为单个冻结图的任何方法。 我要实现的是使用tensorflow Graph Transform Tool应用转换进行优化。不能在saved_model.pb上应用,并且失败并出现以下错误:

failed with Can't parse...both text and binary parsing failed

0 个答案:

没有答案