visdial-rl icon indicating copy to clipboard operation
visdial-rl copied to clipboard

Published 20 hours ago • verified publisher icon batra-mlp-lab

RuntimeError: CUDA error: out of memory

Open Sxx1995 opened this issue 6 years ago • 2 comments

I ran the command "python train.py -useGPU -trainMode sl-abot". However, I got such error "RuntimeError: CUDA error: out of memory". How much GPU memory is needed when training the model? The memory of my GPU is 12212MiB. Thanks!

Sxx1995 avatar Dec 10 '18 12:12 Sxx1995

Hi, I encountered the same problem. Is the problem solved?

Jiaqiang-Ruan avatar Oct 08 '19 15:10 Jiaqiang-Ruan

please wrap the rankABot calling (https://github.com/batra-mlp-lab/visdial-rl/blob/master/train.py#L409) with with torch.no_grad():

sinhprous avatar Apr 05 '21 07:04 sinhprous