chenyilun95 / tf-cpn

Cascaded Pyramid Network for Multi-Person Pose Estimation (CVPR 2018)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A little confuse about "epoch_size".

GengDavid opened this issue · comments

Thanks for your great work!
But I'm a little confuse about epoch_size = 60000 # include flip * 2, aug * 4, batch * 16 in config.py
Could you explain about how to get 60000 a little? We know that COCO has about 150k person instances for training. Is there any connection between these two figures?

Sorry about that. It's from a previous version of code but I forgot to delete the comments. The epoch_size means how many images are trained in one epoch. 60000 is an empirical number. Actually a smaller epoch_size is also acceptable, but in order to get more stable result we just trained longer.

Got it. Thanks for your explanation.