Custom binarization layer with straight through estimator gives error

autograd
5

But the thing is, in the backward pass I want my layer to return the gradients of x clamped between -1 and +1 and not just gradients of x itself :confused:

1 Like