小编Jay*_*Yip的帖子

Tensorflow estimator 多 GPU 预测

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

multi-gpu tensorflow tensorflow-estimator

5
推荐指数
0
解决办法
418
查看次数