2
スリムライブラリにdtype=tf.half
を定義する方法はありますか?一般に、私はfp16
と訓練したい、つまりすべてのテンソルを半精度にしたいとしましょう。これをtf.slim
に指定する方法はありますか? tensorflow使用中の半精度についてテンソルフロースリムと半精度
スリムライブラリにdtype=tf.half
を定義する方法はありますか?一般に、私はfp16
と訓練したい、つまりすべてのテンソルを半精度にしたいとしましょう。これをtf.slim
に指定する方法はありますか? tensorflow使用中の半精度についてテンソルフロースリムと半精度
:tf.float16
例:
import tensorflow as tf import tensorflow.contrib.slim as slim weights = slim.variable('weights', shape=[10, 1], dtype=tf.float16)