Out of memory with length=786
dongshuhao opened this issue · comments
Hey,
Thanks for sharing the code. I am playing the code with activitynet example. I have 1080 ti GPU with 11G memory. With that, I can only run your code with length=196. I just wonder any tricks there to make it run with length=786? I would really appreciate your help!
It seems I have to turn on cudnn in order to make it fit into GPU memory.