Hi, is your code final by now?
And why won’t it be faster?
I have been struggling with similar question recently, I set the data below the threshold to zero and also set the gradient of them to zero. But based on your discussion, I think I should create mask for the loss instead?
So now both my gradient is incorrect and the operation is super slow, any thoughts on that?
Thanks a lot.
My code is:
          
            
           72