search

zhaoweicai / mscnn

Caffe implementation of our multi-scale object detection framework
404 stars 211 forks source link

GPU memory problem #6

Closed lliuz closed 8 years ago

lliuz commented 8 years ago

@zhaoweicai I've download the pretrained model of '8x_768' for test. But it loged a OOM message. I'm using 1 K40 for test, the memory is equal to Titan used in the papar. I try to train the model by myself then, but the OOM proplem occurs again when I using the default batch size, and when I reduce the batch size by half, It seems normal but uses about 10G memory. It's so strange..

lliuz commented 8 years ago

I've solved this problem...

coolhebei commented 8 years ago

I also accounted with this problem. Could you tell me how to solve it? Thank you much very!