冻结TensorFlow Estimator的图形(转换为Core ML)

时间:2018-02-27 20:16:22

标签: tensorflow coreml tensorflow-estimator

我正在尝试将Iris estimator示例转换为与iOS / CoreML一起使用,但无法弄清楚如何导出冻结图。

按照指南ProgrammerMobile指南,我在尝试转换经过训练的(Estimator)graph.pbtxt(和model.ckpt)时出错:

freeze_graph --input_graph=graph.pbtxt --input_binary=false --input_checkpoint=model.ckpt-1000 --output_graph=frozen_graph.pb

给出以下错误:

[...]freeze_graph.runfiles/org_tensorflow/tensorflow/python/training/saver.py", line 683, in _ValidateAndSliceInputs
variable) TypeError: names_to_saveables must be a dict mapping string names to Tensors/Variables. Not a variable: Tensor("dnn/hiddenlayer_0/bias:0", shape=(100,), dtype=float32)

0 个答案:

没有答案