Run demo_coco.py appear Cuda out of memory

Question

Run demo_coco.py appear Cuda out of memory

InstantWindy opened this issue 5 years ago · comments

Hi! Does it take up a lot of memory to test demo_coco files?I don't know that every time I run this file there is always a Cuda out of memory. Thanks!

Jianyuan Wang · Answer 1 · Thu Dec 27 2018 13:13:30 GMT+0800 (China Standard Time)

hi
it will occupy around 10 GB. Please notice the settings if it reports 'out of memory'

InstantWindy · Answer 2 · Thu Dec 27 2018 23:02:30 GMT+0800 (China Standard Time)

Can I ask you a question？When my convolution kernel value is only 1 and -1 or when my input value is only 1 and -1, will the network run faster?

Jianyuan Wang · Answer 3 · Fri Dec 28 2018 00:39:29 GMT+0800 (China Standard Time)

It's a bit weird. Do you mean change the kernel size? If so, you have to re-design the network architecture and cannot use the pretrained model.