Model overfitting - validation not improving

I might be shooting in the dark but maybe it has something to do with this: https://discuss.pytorch.org/t/model-eval-gives-incorrect-loss-for-model-with-batchnorm-layers/7561?

1 Like