search

bingykang / Fewshot_Detection

Few-shot Object Detection via Feature Reweighting
https://arxiv.org/abs/1812.01866
526 stars 111 forks source link

CUDA out of memory #23

Open ghost opened 4 years ago

ghost commented 4 years ago

I have finished trained the model, but when I tried to evaluate the model, it printed out 'CUDA out of memory'. I have 2 GPUs and 32GB. But I can't use 2GPUs to evaluate. It evaluated with 1GPU every time even though I have changed gpu to "0,1" in valid_ensemble.py. Do you have any suggestion or solution?

CHS18895660306 commented 3 years ago

hello,I get this question, too. Have you solve this problem?

lander1003 commented 3 years ago

Do you solve the problem?

aeinsara commented 3 years ago

you should reduce batchsize.

MartinYYYYan commented 2 years ago

@aeinsara Hi! I have tried to decrease the batch size, but It is not working. Have you solved this problem?

MusherM commented 2 years ago
  1. reduce batchsize
  2. update your graphic card(lol)
  3. disable random
  4. increase subdivisions
GDDKRJ commented 1 year ago

hello,I met this same question, too. Have you solved this problem?