Detectron2里的“epoch”如何设置？

Question

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

Answer 1

以下为解答，望采纳

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