Closed superxi closed 3 years ago
Hi : I‘m wondering how much GPU memory is needed for training with batch size 1. I'm training with 2080Ti 11G, It always CUDA out of memory even when batch size is 1
try ReLU(inplace=True)
I have a same problem. Did you solved this problem?
Hi : I‘m wondering how much GPU memory is needed for training with batch size 1. I'm training with 2080Ti 11G, It always CUDA out of memory even when batch size is 1