Where do you see the number of epochs set?

[zhxiru997]

Please, where do you see the number of epochs set? What standard does the training end with? Because I see that there is no epoch setting in the training parameters, I don't know how many epochs will end after training.

[sharvil]

There's no explicit limit on the number of epochs; rather the limit is on total number of training steps (minibatch iterations). You can specify the number of steps from the command line, e.g. python -m wavegrad /path/to/model/dir /path/to/dir/containing/wavs --max_steps=100000 or similar.

You can always make changes to this code if you want to terminate on an epoch boundary.