Size mismatch between weights and momentums.

Just chance it ran in the first place I think.