Search code examples
iterationobject-detectiondetectron

How many images per iteration in Detectron2


I am new to using detectron2, just learning it.This might be a noob question, but I really need the answer. I find nothing related to number of epochs in the repository.

We know , epoch means single passing of all data through the model, and a batch means a certain subset of the whole dataset, that has the ability to impact the loss through gradient descent. In this model, (Detectron2) we have something called iteration. What does this iteration mean? Does it mean passing of one batch through the mode, or an epoch(that shouldn't be the case considering time per iteration)

My question is , how do I know the minimum number of iterations that will pass all my images to the model, at least once.


Solution

  • In detectron2, epoch is MAX_ITER * BATCH_SIZE / TOTAL_NUM_IMAGES