about the parameters of train.py

Open tianguang2525 opened this issue 2 years ago • 1 comments

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.

Mar 21 '23 12:03 tianguang2525

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.

Apr 04 '23 07:04 DavidHuji