Open dagnichz opened 5 years ago
i'm trying to do guided backpropogation on mobilenet using your code with tensorflow 1.14. unfortunately, tensorflow does not reposnd to the override you suggested:
@ops.RegisterGradient("GuidedRelu") def _GuidedReluGrad(op, grad): return tf.where(0. < grad, gen_nn_ops.relu_grad(grad, op.outputs[0]), tf.zeros(grad.get_shape()))
I also tried chganing the override line to "Relu6" but still nothing. with graph.gradient_override_map({'Relu6': 'GuidedRelu'}):
mybe you tried doing it yourself on mobilnet and can offer a solution?
i'm trying to do guided backpropogation on mobilenet using your code with tensorflow 1.14. unfortunately, tensorflow does not reposnd to the override you suggested:
@ops.RegisterGradient("GuidedRelu") def _GuidedReluGrad(op, grad): return tf.where(0. < grad, gen_nn_ops.relu_grad(grad, op.outputs[0]), tf.zeros(grad.get_shape()))
I also tried chganing the override line to "Relu6" but still nothing. with graph.gradient_override_map({'Relu6': 'GuidedRelu'}):
mybe you tried doing it yourself on mobilnet and can offer a solution?