Using SparseTensor as a trainable variable?

前端 未结 4 1944
借酒劲吻你
借酒劲吻你 2020-12-09 13:39

I\'m trying to use SparseTensor to represent weight variables in a fully-connected layer.
However, it seems that TensorFlow 0.8 doesn\'t allow to use Sp

4条回答
  •  情话喂你
    2020-12-09 14:19

    The above code works with some minor correction like this.

    def optimize(loss, mask_tensor):
        optimizer = tf.train.AdamOptimizer(0.001)
        grads_and_vars = optimizer.compute_gradients(loss)
        modified_grads_and_vars = [
            (tf.multiply(gv[0], mask_tensor[gv[1]]), gv[1]) for gv in grads_and_vars
        ]
        return optimizer.apply_gradients(modified_grads_and_vars)
    

提交回复
热议问题