Out of memory with length=786

VisionLearningGroup / R-C3D

code for R-C3D

MIT License

254 stars 94 forks source link

Out of memory with length=786 #25

Closed dongshuhao closed 6 years ago

dongshuhao commented 6 years ago

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!

dongshuhao commented 6 years ago

It seems I have to turn on cudnn in order to make it fit into GPU memory.