TensorFlow estimator加载pb模型预测TfRecord中的样本
TensorFlow estimator加载pb模型预测TfRecord中的样本
最近在上一个模型,需要测试模型输出与线上server predict值的一致性,模型用到了高级API estimator以及tf record格式的数据,很多地方还不是很熟悉。为了测一致性我构造了一个只有一条样本的数据,然后想在本地加载模型export出来的pb格式文件,再load 数据输出预测值。在我的场景下是要捞...
原创
2020-03-16 21:01:32 ·
3741 阅读 ·
1 评论