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

Question

tensorflow.contrib.slim中的新学习模块看起来很有前途： https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/slim/python/slim/learning.py 我试图弄清楚如何在我只有一个工作节点但有几个GPU的配置上使用这个新模块重现CIFAR 10 multi-gpu示例（或ImageNet示例）。

Answer 1

我使用https://github.com/tensorflow/models/tree/master/slim/deployment取得了一些成功 创建配置对象时，您需要设置num_clones = [num_gpus]

例如，

config = model_deploy.DeploymentConfig(num_clones=2)

查看model_deploy.py文件中的示例。