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)