How to set epoch in training

[Valkyrie1215]

Hi，
Thank you for open-sourcing this amazing work! I encountered a problem while trying to replicate it: I couldn't find where to set the number of epochs during the training process. Could you please guide me on this? Thanks

[AniketGurav]

@Valkyrie1215 can you please share the model weights if you have downloaded it from baidu.