how to reduce the training number on each epoch

[Lvhhhh]

the training number in your code is 400000 one epoch .it cost too much time . how to reduce it?

[avalonstrel]

It depends on the dataset, but you can modify the source code to make a short cut.