Tensorflow估计器多GPU预测

时间:2019-01-02 08:32:11

标签: tensorflow tensorflow-estimator multi-gpu

我有一个估算器,可以通过在配置中指定train_distributeeval_distribute来使用多个GPU进行训练和评估。但是predict_distribute中没有tf.estimator.RunConfig参数。如何预测使用多个GPU?

非常感谢。

0 个答案:

没有答案