Detectron2里的“epoch”如何设置?

正在用Detectron2训练Mask RCNN,了解到Detectron2里并没有“epoch”这个概念,那Detectron2是如何实现“epoch”的呢?

以下为解答,望采纳

Detectron2没有epoch参数,在使用Detectron2训练Mask RCNN时,可以设置参数SOLVER.MAX_ITER来控制训练的迭代次数,间接地控制训练的"epoch"数量。例如,如果将SOLVER.MAX_ITER设为1000,则意味着模型会在训练数据集上进行1000次迭代,即训练1000个"mini-batch",在总共1000个batch的数据上等价于训练1个"epoch"。