A question about InverseTransform2D

[ghost]

Hi,
In your code, the class InverseTransform2D calculates the loss of boundary distance. You use "inputs = torch.ge(inputs, 0.5).float()" to get boundary map. But it seems that the backward is interrupted. Please make sure that this item of loss can be backward. Cause I think the required_grad attribute of mean_square_inverse_loss is False.

[sborse3]

Thanks for pointing it out, an earlier bug had creeped in. I've fixed it.