Yuheng

Hi, I have the same problem. "CUDA OUT OF MEMORY", when batch_size > 1. and get "RuntimeError: cuda runtime error", when batch_size=1. I am using the NVIDIA 2080ti with 11GB...

I set the "cudnn.benchmark = False", solve the "CUDA runtime error".