tf.slimで、依存関係を損失に追加する必要があるかどうか

Question

tf.slimでは、私はbatch_normを使用しました。

私の質問は、明示的に損失に依存関係を追加する必要があるかどうかです。

私は、スリムは、自動的に損失に依存関係を追加しているかどうか、私はbatch_normを使用して知っていたと思いますか？私は非常に混乱しています。

Answer 1

はい、必要です。

注：

あなたは指示に従ってくださいhereでしトレーニング、moving_meanとmoving_varianceを更新する必要があるとき。デフォルトでは、更新操作はtf.GraphKeys.UPDATE_OPSに配置されているため、train_opへの依存として追加する必要があります。例：

update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS) 
with tf.control_dependencies(update_ops): 
    train_op = optimizer.minimize(loss)