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?
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?