Hi,
Yes your self.fc1 and self.fc2 layers will never get any gradients here.
The operation out_ticket > 0.2 is not differentiable as the gradients are 0 almost everywhere. You won’t be able to do masking in a differentiable way.