Open ghost opened 4 years ago
hello,I get this question, too. Have you solve this problem?
Do you solve the problem?
you should reduce batchsize.
@aeinsara Hi! I have tried to decrease the batch size, but It is not working. Have you solved this problem?
hello,I met this same question, too. Have you solved this problem?
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?