Masked loss function (solved)


(Bodo Kaiser) #1

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