标签： tensorflow
我正在使用slim.batch_norm，我想使用float16训练，但移动平均值在float32类型中更新。如何在float16中更新它们？如果我将它们添加到tf.GraphKeys.TRAINABLE_VARIABLES，有什么办法将所有tf.GraphKeys.TRAINABLE_VARIABLES强制转换为float16？