Hi guys,
In version 0.2.0_post3 I used to do the following operation to remove NaNs that were being produced by a division by zero:
out[out != out] = 0
and it used to work!
With version 0.3.0_post4 I’m getting the following error message:
assert not ctx.needs_input_grad[1], "MaskedFill can't differentiate the mask"
AssertionError: MaskedFill can't differentiate the mask

Any tips / workarounds?
Cheers,
Miguel