1

tensorflow multi-gpu CIFAR 10 exampleでは、GPU ごとに損失を計算します (行 174-180 ) 。

for i in xrange(FLAGS.num_gpus):
  with tf.device('/gpu:%d' % i):
    with tf.name_scope('%s_%d' % (cifar10.TOWER_NAME, i)) as scope:
      loss = tower_loss(scope)

数行下(246行目)の場合、次のように評価さlossれます

_, loss_value = sess.run([train_op, loss])

どの損失が正確に計算されますか?

関数を確認しましtower_lossたが、すべての GPU (タワー) での増分集計は見られません。

グラフ全体が (すべての GPU で) 実行されていることは理解していますが、返される損失の値は何ですか? loss最後の GPUのみ?loss実際の変数に集計が表示されません。

4

2 に答える 2