CPU and GPU memory consumption during inference

[GeorgePearse]

Original issue: open-mmlab/mmdetection#1170
State in upstream: OPEN
Created: 2019-08-12
Last updated: 2021-04-19

---

DOUBT

While running the demo notebook for MaskRCNN model. I want to know why the script uses :

1. Only 1.4 GB GPU memory ?
2. 2.95 GB CPU memory ?
3. The notebook loads more than 20 CPU threads ?
4. Is there a way to control the total CPU threads and memory?

STEPS TO REPRODUCE

config_file = '../configs/mask_rcnn_r50_fpn_1x.py'
checkpoint_file = '../checkpoints/mask_rcnn_r50_fpn_2x_20181010-41d35c05.pth'
model = init_detector(config_file, checkpoint_file, device='cuda:0')
result = inference_detector(model, img)