What is the relationship between nproc_per_node, SOLVER.IMS_PER_BATCH, iterations, epoches and number of training images?

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

https://arxiv.org/abs/1904.01355

Other

3.28k stars 630 forks source link

What is the relationship between nproc_per_node, SOLVER.IMS_PER_BATCH, iterations, epoches and number of training images? #239

Open becauseofAI opened 4 years ago

becauseofAI commented 4 years ago

for example: nproc_per_node=8 SOLVER.IMS_PER_BATCH=16 iterations=90k num_train_ims=118287 (coco2017)

========================= batch_size = SOLVER.IMS_PER_BATCH = 16 ? ims_per_device = SOLVER.IMS_PER_BATCH / nproc_per_node = 16 / 8 = 2 ? epochs = iterations iterations / num_train_ims = 90k 16 / 118287 = 12.17 = 13 ?