watsonyanghx / CNN_LSTM_CTC_Tensorflow

CNN+LSTM+CTC based OCR implemented using tensorflow.
MIT License
362 stars 210 forks source link

关于batch normalization层 #18

Open huyi1989 opened 6 years ago

huyi1989 commented 6 years ago

如果想要保存跟新moving_mean和moving_variance的话,好像要写以下代码: extra_update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS) with tf.control_dependencies(extra_update_ops): train_op = optimizer.minimize(loss) 但是好像没找到相关内容,想问作者是不是漏了写了?

watsonyanghx commented 6 years ago

这里设置了updates_collections=None,可以看一下tf.contrib.layers.batch_norm