无法冻结来自tf.estimator的经过训练的模型

时间:2018-08-14 15:32:13

标签: tensorflow tensorflow-estimator

我尝试使用freeze_graph.py脚本冻结graph.pbtxt和model.ckpt-2000,但是在下面出现错误。

  

TypeError:names_to_saveables必须是将字符串名称映射到张量/变量的字典。不是变量:Tensor(“ dnn / hiddenlayer_0 / bias:0”,shape =(10,),dtype = float32)

这是我使用的命令。

  

python Frozen_graph.py --input_graph = graph.pbtxt --input_checkpoint = model.ckpt-2000 --output_graph = frozen_graph.pb --output_node_names = output_node --input_binary = false

该模型来自此页面中的示例。 https://www.tensorflow.org/versions/r1.3/get_started/estimator

  

feature_columns = [tf.feature_column.numeric_column(“ x”,shape = [4])]

     

classifier = tf.estimator.DNNClassifier(feature_columns = feature_columns,hidden_​​units = [10,20,10],n_classes = 3,model_dir =“ / tmp / iris_model”)

知道为什么它不起作用吗?

0 个答案:

没有答案