tensorflow / tensorflow/models
A tool to profile RAM and GPU memory when working with tensorflow object detection API
Open
@pkulzc is already working on this.
Since Jul 7, 2020.
models:research:odapi
type:feature
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
Prerequisites
1. The entire URL of the file you are using
https://github.com/tensorflow/models/tree/master/research/object_detection
2. Describe the feature you request
I am struggling with OOM errors with the tensorflow object detection API. The error happens even when I use compressed data.
So I need to know if there's a way to profile memory while training an object detector with the tensorflow object detection API. That would help us understand what is slowing down training and causing OOM errors.
3. Additional context
Add any other context about the feature request here.
4. Are you willing to contribute it? NO
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.