Feature extraction: CUDA out of memory
Open
- Dominant language
- Python
- Stars
- 733
- Forks
- 176
- PR merge metrics
- No merged PRs in 30d
Description
Hi,
I am trying to extract features for coco dir - however it always gives me RuntimeError: CUDA out of memory.
I have tried using single GPU by setting CUDA_VISIBLE_DEVICES=0- still ti gives me the same Runtime error even when the GPU is free.
What should I do?
Regards,
Vedika
Contributor guide
No contributing guide indexed for this repository
Research direction
Start at the feature-extraction entry point for the COCO directory and reproduce the reported CUDA out-of-memory error with CUDA_VISIBLE_DEVICES=0. Check the available configuration and runtime memory behavior; done means feature extraction completes successfully without the reported error.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100