Detectron2: About batch size

Created on 30 Apr 2020  ·  1Comment  ·  Source: facebookresearch/detectron2

❓ What the parameter "cfg.MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE" is used for? I understood that it doesn't identify the training batch size but from your docs I do not understand its meaning

picture Ilbotre

Most helpful comment

cfg.MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE parameter is used to sample a subset of proposals coming out of RPN to calculate cls and reg loss during training. Calculating loss on all RPN proposals isn't computationally efficient.

picture Kbkartik on 30 Apr 2020
👍3

>All comments

cfg.MODEL.ROI_HEADS.BATCH_SIZE_PER_IMAGE parameter is used to sample a subset of proposals coming out of RPN to calculate cls and reg loss during training. Calculating loss on all RPN proposals isn't computationally efficient.

Kbkartik picture Kbkartik on 30 Apr 2020
👍3
Was this page helpful?
0 / 5 - 0 ratings

Related issues

jinfagang picture jinfagang  ·  3Comments
guy4261 picture guy4261  ·  4Comments
kl720 picture kl720  ·  3Comments
soumik12345 picture soumik12345  ·  3Comments
marcoippolito picture marcoippolito  ·  4Comments