def forward(self, input, target):
    input = input.contiguous()
    target = target.contiguous()
    target = torch.where(torch.isnan(target), input, target)
    if (input.size(0) > 0):
        loss = self.iou3d_loss(input, target).sum()
    else:
        loss = ((input - target).sum(1) * 0.0)
    return loss
def forward(self, input, target):
    input = input.contiguous()
    target = target.contiguous()
    target = torch.where(torch.isnan(target), input, target)
    if (input.size(0) > 0):
        loss = self.iou3d_loss(input, target).sum()
    else:
        loss = ((input - target).sum(1) * 0.0)
    return loss