In the python implementation's README-file there is no mention of how to change the number of epochs, e.i. "--epoch_max ...."