张量流量细长半精度

Question

有没有办法在slim库中定义dtype=tf.half？一般来说，假设我想用fp16进行训练，即希望所有张量都精确到半精度。有没有办法将其指定为tf.slim？

Answer 1

对于张量流使用的半精度：tf.float16
例如：

import tensorflow as tf
import tensorflow.contrib.slim as slim
weights = slim.variable('weights', shape=[10, 1], dtype=tf.float16)