about the parameters of train.py

[tianguang2525]

Hello, thanks for sharing the wonderful code!
Could you please tell me how to configure the parameters of the train.py file to obtain the effect of the published model, or how many epochs of training are required.

[DavidHuji]

Hi, thanks for the comment.
If you follow the instructions in the Readme, the default configuration should be sufficient.
Regarding the number of epochs, we trained until the validation results started to get worse. For COCO it was about 8 epochs but for smaller datasets (i.e. FlickrStyle10k), it was about 25 epochs.