尝试在Tensorflow 1.3中冻结图形时,冻结工具中的TypeError

时间:2017-10-13 18:36:44

标签: tensorflow

我在Python中训练tf.contrib.learn估算器(特别是DNNLinearCombineRegressor),并在定义估算器时通过指定model_dir来保存模型的参数和图形。训练完成后,我尝试使用此post中提到的CLI冻结图表,并收到此错误:

TypeError:names_to_saveables必须是字符串映射到Tensors / Variables的字符串名称。不是变量:Tensor(" dnn / hiddenlayer_0 / biases:0",shape =(10,),dtype = float32)

知道如何解决这个问题吗?另外,我如何确保在训练之后,当我使用估计器的predict_scores()进行预测时,冻结的图形文件用于创建模型?我想冻结图形,以便predict_scores()不会每次重新加载图形及其变量以进行预测。提前谢谢。

0 个答案:

没有答案