Masked loss function (solved)

vision
1

I somehow accidentally deleted optimizer.zero_grad() which caused the problems.