I used BatchNorm1d() with my 4-layer net arch on MNIST dataset. I added accuracy estimation to training function (model.train() mode) and it shows loss/accuracy - 0.49/0.48 after couple epochs. And for evaluate function on dev set it shows - 4.26/0.0019. Is it that “a bit to high” estimate during training or it looks more like my arch errors?