Resnet50 out of memory

[qinhaifangpku]

hi~ I am run your code in my machine which is 12G titanx. datasets is Coco. when I am traing the resnet 50 +3stage, It could not work, It will out of memory. Could you please tell me how do you train resnet101+5stage on a 12G titanx? thank you for your help in advance!

[hgaiser]

What network are you running exactly? I believe the authors never released their resnet implementation. Did you implement something yourself? If so, it'd be good to have the prototxt of your network so it becomes more clear what the issue is.

[qinhaifangpku]

thank you！ I just replace the VGG with the resnet50