I have reproduced your code in my scenario, but after iterations, the loss becomes negative. Is this normal?