如何在Android / IOS中使用tf.train.ExponentialMovingAverage

时间:2017-06-22 04:48:02

标签: android python tensorflow tensorflow-serving tensorflow-gpu

我使用(from pap in db.Papers where authers.Split(',').ToList().Contains((pap.auther) select pap).ToList(); 将我的模型导出到名为freeze_graph的文件中。但发现"frozen.pb"预测的准确性非常糟糕。

我知道问题可能frozen.pb未包含在MovingAverage中。

当我使用frozen.pb文件恢复模型以进行评估时,如果我调用model.ckpt,那么准确性就像预期的那样好,否则准确性很差。

那么如何导出二进制模型的性能与从检查点文件恢复的性能相同?我想在Android设备中使用tf.train.ExponentialMovingAverage(0.999)个文件。

The official document没有提到这一点。

谢谢!

冻结命令:

".pb"

评估代码:

~/bazel-bin/tensorflow/python/tools/freeze_graph \
  --input_graph=./graph.pbtxt \
  --input_checkpoint=./model.ckpt-100000 \
  --output_graph=frozen.pb \
  --output_node_names=output  \
  --restore_op_name=save/restore_all \
  --clear_devices

0 个答案:

没有答案