Getting back 0 std messages from own data

[SalmonPenny]

Dear @MilesCranmer and team,

I am using this repo for a school project and am experiencing an annoying issue. When I implement my own data, I sometimes get back 0 std messages. All e(number) messages have the same value per column. I've just ran the same code twice in a row and once I got back all 0's and now I do get back non-zero numbers.

These are from small tests with only 1 epoch, but when I ran the code for 50 epochs earlier I got back 0's. This is very annoying since I always have to wait a long time for the code to finish.

Is there something I could do to avoid this issue? Maybe something to do with the data itself? Or a hyperparameter?

Best regards,
L.W.