tensorflow / tensorflow/models
Training LSTM Object Detection on multiple GPUs
Open
@yongzhe2160 is already working on this.
Since Mar 12, 2021.
models:research
type:docs
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
Prerequisites
Please answer the following question for yourself before submitting an issue.
- [ x] I checked to make sure that this issue has not been filed already.
1. The entire URL of the documentation with the issue
https://github.com/tensorflow/models/tree/master/research/lstm_object_detection
2. Describe the issue
I would like to train the model on a single machine with multiple GPUs. Will setting the worker_replicas variable in train.py to the number of GPUs do the job, or is there something else I need to do? Thanks!
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.