The epoch argument is wrong

[wlnirvana]

The --n_epochs argument is in fact an iteration, which updates the weight based on a random sample of data points.

There is no conclusive definition, but the community generally define epoch to be an entire pass of the whole dataset. See this and this

[samlanka]

Yes, the code is counting the number of iterations, not number of epochs