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)
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)