[latenightcoding]

This is a thinly veiled "machinebox" ad. Thanks for teaching us how to overfit in under an hour OP.

[minimaxir]

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