我有一个估算器,可以通过指定train_distribute和eval_distribute在配置中使用多个 GPU 进行训练和评估。但predict_distribute在tf.estimator.RunConfig. 如何使用多个 GPU 进行预测?
train_distribute
eval_distribute
predict_distribute
tf.estimator.RunConfig
multi-gpu tensorflow tensorflow-estimator
multi-gpu ×1
tensorflow ×1
tensorflow-estimator ×1