facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
Apache License 2.0
26.22k stars 5.45k forks source link

Validation: out of memory #910

Open lssam opened 5 years ago

lssam commented 5 years ago

I am trying to validate detection over ~87k classes with an FPN-faster-rcnn-101-2x on a validation set of 10000 Images. I keep getting an out of memory error, because of the inefficient way of validating in detectron with: https://github.com/facebookresearch/Detectron/blob/a6a835f5b8208c45d0dce217ce9bbda915f44df7/detectron/core/test_engine.py#L383

is there another option to validate with a memory friendly fashion ?