DetectionTeamUCAS / DetectionTeamUCAS/Faster-RCNN_Tensorflow

decreasing the amount of ram used on gpu

Open
#26 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
149
Forks
61
PR merge metrics
No merged PRs in 30d

Description

Hello,

I'm trying to decrease the amount of ram used on the gpu for inference.

i've used the supplied cfgs.py that uses resnet101,
it uses up about 6G during training,
then i've frozen the graph -> created a 190MB frozen file,
i then use this frozen file from CPP , it consumes about 800MB on gpu for inference.

i then tried using the cfgs_res50.py, this lead to 3.5G of ram using on gpu during training,
a 115MB frozen graph file, and ram consumed on gpu during inference (from cpp) has not decreased significantly ( about 720M).

can you suggest a method to decrease the ram consumption on gpu ?
( maybe some other property of the network ? )

thanks,
Omer.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by comparing cfgs.py and cfgs_res50.py, then inspect how the frozen graph is loaded and run from C++. Measure GPU RAM during training and inference for both graphs, using the reported 800 MB and 720 MB figures as a baseline. Done would require a demonstrated way to reduce inference GPU usage, but the issue does not define a target or specific implementation area.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, tensorflow
Domain
computer-vision, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.