tensorflow苗条和半精度

Question

有没有一种方法来定义dtype=tf.half苗条的图书馆？一般来说，我们可以说我想用fp16进行训练，即希望所有张量都是半精度。有没有办法指定这个到tf.slim？

Answer 1

对于tensorflow使用半精度：tf.float16
实施例：

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