Tensorflow 使用tensorfow.slim.learning进行多GPU培训

Tensorflow 使用tensorfow.slim.learning进行多GPU培训,tensorflow,tf-slim,Tensorflow,Tf Slim,tensorflow.contrib.slim中的新学习模块看起来非常有前途: 我试图弄清楚如何在只有一个工作节点但有多个gpu的配置上使用此新模块重现CIFAR 10多gpu示例(或ImageNet示例)。我在使用 创建配置对象时,您将设置num\u clones=[num\u gpus] 比如说, config = model_deploy.DeploymentConfig(num_clones=2) 查看model_deploy.py文件中的示例

tensorflow.contrib.slim中的新学习模块看起来非常有前途:
我试图弄清楚如何在只有一个工作节点但有多个gpu的配置上使用此新模块重现CIFAR 10多gpu示例(或ImageNet示例)。

我在使用 创建配置对象时,您将设置num\u clones=[num\u gpus]

比如说,

config = model_deploy.DeploymentConfig(num_clones=2)
查看model_deploy.py文件中的示例