[minimaxir]

I suspect the OP cheated by only training on 1 epoch; otherwise, the overfit would become obvious as the validation accuracy craters.