Cadene / Cadene/vqa.pytorch

Feature extraction: CUDA out of memory

Open
#46 1 comment 0 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.