我正在尝试将Iris estimator示例转换为与iOS / CoreML一起使用,但无法弄清楚如何导出冻结图。
按照指南Programmer和Mobile指南,我在尝试转换经过训练的(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)