如果使用高级估算器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