标签: tensorflow tensorflow-estimator multi-gpu
我有一个估算器,可以通过在配置中指定train_distribute和eval_distribute来使用多个GPU进行训练和评估。但是predict_distribute中没有tf.estimator.RunConfig参数。如何预测使用多个GPU?
train_distribute
eval_distribute
predict_distribute
tf.estimator.RunConfig
非常感谢。