Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

RutimeError:CUDA out of memory. when i training on COCO train2017 dataset #32

Open
houshuaishuai opened this issue Mar 4, 2022 · 2 comments
Open

RutimeError:CUDA out of memory. when i training on COCO train2017 dataset #32

houshuaishuai opened this issue Mar 4, 2022 · 2 comments

Comments

@houshuaishuai
Copy link

when i run python tools/train.py --cfg experiments/coco/w32/w32_4x_reg03_bs10_512_adam_lr1e-3_coco_x140.yaml
RuntimeError: CUDA out of memory. Tried to allocate 86.00 MiB (GPU 0; 10.91 GiB total capacity; 9.39 GiB already allocated; 125.50 MiB free; 9.48 GiB reserved in total by PyTorch) , i didn't find batch_size in the w32_4x_reg03_bs10_512_adam_lr1e-3_coco_x140.yaml? is there anything else that need to be modified?
@AgntBrwr
Copy link

AgntBrwr commented Jun 1, 2022

Images per GPU (for training) can be lowered in that file as a solution

@wusaisa
Copy link

wusaisa commented Oct 30, 2023

I had the same problem as you, did you solve it?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@houshuaishuai @wusaisa @AgntBrwr