您的位置:首页 > 其它

InvalidArgumentError : Shape in shape_and_slice spec [,] does not match the shape stored in checkpoi

2018-02-02 16:34 701 查看
解决方法:

classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,hidden_units=[10, 20, 10],n_classes=2,

                                              model_dir="/tmp/iris_model")

这里的model_dir给改了。。因为tensorflow默认会先去找已经训练过的模型

tensorflow小白伤不起。。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  tensorflow spec checkpoint
相关文章推荐