facebookresearch / facebookresearch/perception_models
OOM Error when Running DETA_PE Evaluation Script with 3x RTX 4090D (24GB) GPUs
Open
PE
- Dominant language
- Jupyter Notebook
- Stars
- 2.4k
- Forks
- 162
- PR merge metrics
- No merged PRs in 30d
Description
Description:
When executing scripts/eval_1824pix.sh to evaluate the deta_coco_1824pix.pth model on COCO data using 4 x RTX 4090D GPUs (24GB each), I encounter an Out-of-Memory (OOM) error. Could you suggest which parameters to adjust to reduce GPU memory usage during inference?
Key Context:
Hardware: 4× NVIDIA RTX 4090D (24GB VRAM)
Script: scripts/eval_1824pix.sh
Model: deta_coco_1824pix.pth
Contributor guide
Assessment
This issue has not been assessed yet.