How many GPUs at least to use when training?

YueLiao / PPDM

Code for "PPDM: Parallel Point Detection and Matching for Real-time Human-Object Interaction Detection".

MIT License

219 stars 42 forks source link

Closed SherlockHolmes221 closed 4 years ago

SherlockHolmes221 commented 4 years ago

I got OOM when using 2 GPUs(--gpus 2,3), I want to know how many GPUs at least to use when training? Looking forward to your reply, thanks.

SherlockHolmes221 commented 4 years ago

solve it by decreasing the batch_size, close it