Fix torch.range warning in the DilateLoss

[isaacmg]

There is a new warning about using torch.range with respect to the Dilateloss

[isaacmg]

The fix for this pretty easy. Essentially just replace with torch.arrange() This function is deprecated and will be removed in a future release because its behavior is inconsistent with Python’s range builtin. Instead, use torch.arange(), which produces values in [start, end).