solver.py 128, 134 .. line - Before (Now) d1_loss = torch.mean((out-1)**2) - After d1_loss = torch.mean((out-1).clamp(max=0)**2)