在张量流中没有为操作 MaxPoolWithArgmax 定义梯度 [英] No gradient defined for operation MaxPoolWithArgmax in tensorflow

问题描述

我想在 TensorFlow 中使用 max_pool_with_argmax 操作，但出现以下错误:

I wanna use max_pool_with_argmax operation in TensorFlow, but I got following error:

LookupError: 没有为操作 'MaxPoolWithArgmax_1' 定义梯度(操作类型:MaxPoolWithArgmax)

LookupError: No gradient defined for operation 'MaxPoolWithArgmax_1' (op type: MaxPoolWithArgmax)

这是我使用的一段代码max_pool_with_argmax:

Here is a piece of my code that I am using max_pool_with_argmax:

BN_relu13 =   tf.nn.relu(tf.nn.batch_normalization(h_conv13,batch_mean13,batch_var13,tf.Variable(tf.zeros([64])),tf.Variable(tf.ones([64])),epsilon)) 
#max pooling 
h_pool1, argmax_1 = max_pool_2x2(BN_relu13)

这里是 max_pool_2x2 模块:

def max_pool_2x2(x):
    return tf.nn.max_pool_with_argmax(x, ksize=[1, 2, 2, 1],strides=[1, 2, 2, 1], padding='SAME')

在我建立我的网络之后，我使用 AdamOptimizer 来训练我的网络.

and after I built my network I am using AdamOptimizer to train my network.

顺便说一下，当我使用 tf.nn.max_pool(而不是 tf.nn.max_pool_with_argmax)时，一切正常.

By the way, when I am using tf.nn.max_pool (instead of tf.nn.max_pool_with_argmax) everything is working fine.

我在 GPU 上运行我的代码，我在 ubuntu14 上使用 python 2.7.

I am running my code on GPU and I am using python 2.7 on ubuntu14.

谢谢，阿里

推荐答案

梯度 op 实际实现了 此处，但不知何故未注册 此处.值得在其上提交 github 问题.

The gradient op is actually implemented here, but somehow not registered here. Would be worth filing a github issue on it.

同时，您可以按照此教程自行注册 op只有梯度配准部分).

At the same time, you can register the op by yourself following this tutorial (only the gradient registration part).

这篇关于在张量流中没有为操作 MaxPoolWithArgmax 定义梯度的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！