tensorflow - tensorfow.slim.learning を使用した複数の GPU トレーニング

Question

tensorflow.contrib.slim の新しい学習モジュールは非常に有望に見えます : https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/slim/python/slim/learning.pyこの新しいモジュールを使用して、ワーカー ノードが 1 つしかなく、GPU が複数ある構成で、CIFAR 10 マルチ GPU の例 (または ImageNet の例) を再現するにはどうすればよいですか。

Answer 1

https://github.com/tensorflow/models/tree/master/slim/deploymentを使用していくつかの成功を収めまし た config オブジェクトを作成するときに、num_clones = [num_gpus] を設定します

例えば、

config = model_deploy.DeploymentConfig(num_clones=2)

model_deploy.py ファイルの例を確認してください。