Training epochs in fine-tuning stage

Verg-Avesta / CounTR

CounTR: Transformer-based Generalised Visual Counting

https://verg-avesta.github.io/CounTR_Webpage/

MIT License

92 stars 9 forks source link

Closed littlepenguin89106 closed 1 year ago

littlepenguin89106 commented 1 year ago

Hi, How many epochs did you train in fine-tuning stage? The paper only mentioned 300 epochs in pre-train stage.

Verg-Avesta commented 1 year ago

Sorry we didn't mention it in the paper.

We fine-tuned for 1000 epoches and the best result occurs at about 850~950 epoches.