2016-06-18 6 views
2

tensorfow.slim.learningとトレーニング: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/slim/python/slim/learning.py 私はCIFAR 10マルチGPU例を再現できるかを把握しようとしている(またはImageNetの例)この新しいモジュールを使用して、私は1つのワーカーノードしか持たないが、いくつかのGPUを持つ構成にしている。マルチGPU tensorflow.contrib.slimにおける新しい学習モジュールは非常に有望に見える

答えて

2

私はたとえば、[num_gpus] = configオブジェクトを作成するときに、あなたがnum_clonesを設定しますhttps://github.com/tensorflow/models/tree/master/slim/deployment を使用して

をいくつかの成功を持っていた、

config = model_deploy.DeploymentConfig(num_clones=2) 

はmodel_deploy.pyファイルの例をご覧ください。

関連する問題