search

JosephKJ / OWOD

(CVPR 2021 Oral) Open World Object Detection
https://josephkj.in
Apache License 2.0
1.03k stars 155 forks source link

GPU Memory #87

Closed hero-y closed 2 years ago

hero-y commented 2 years ago

When I train with 8 GPUs, gpu1-gpu7 occupies about 5000M memory, but gpu0 occupies about 16000M memory. Is there a problem?

Looking forward to your reply！

JosephKJ commented 2 years ago

The code doesnt explicitly handle the GPU allocation, @hero-y. This is done by PyTorch.