Megvii-BaseDetection / Megvii-BaseDetection/YOLOX
cuda显存一直超出的问题/Persistent cudA video memory overruns
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 10.6k
- Forks
- 2.5k
- PR merge metrics
- No merged PRs in 30d
Description
我目前训练yolox_l，batch_size设置的为4，img_size为416,但一直报显存的错，我的显卡是xp，显存12个g，应该不会出现这样的情况，之前yolov4,v5都设置的batchsize为8或者6。。。，请问有人知道是什么原因吗？
I am currently training yolox_L, set batch_size to 4, imG_size to 416, but the error of the video memory is always reported. My video card is XP, and the video memory is 12 G, this should not happen, yolov4, V5 set batchsize to 8 or 6 before... Does anybody know why?
Contributor guide
No contributing guide indexed for this repository
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.
Research direction
No source file, test, or entry point is named. Start by reproducing YOLOX_L training with batch_size 4 and img_size 416 on the reported 12 GB XP GPU, then capture the complete CUDA out-of-memory error and environment details. Done means identifying a reproducible cause or documenting the missing information needed to diagnose it.
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