yunjey/mnist-svhn-transfer

Critical Issue (Loss Function Error)

Joo-Hyun-Lee opened this issue · 0 comments

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)