ResNet icon indicating copy to clipboard operation
ResNet copied to clipboard

Published 20 hours ago • verified publisher icon tornadomeet

I can't train the resnet-50 on Titan-X with batchsize 256.

Open KangHsi opened this issue 8 years ago • 1 comments

resnet 50 with bachisize 64 takes 8G+ memory I can't train the resnet 50 on my 12G Titan X even with memongers. I follow all the instructions in readme.md, and this seems differing a lot from the argue. The error below arises when I train the net

mxnet.base.MXNetError: [16:09:57] src/storage/./pooled_storage_manager.h:80: cudaMalloc failed: out of memory

KangHsi avatar Mar 07 '17 09:03 KangHsi

sorry, i'm not sure what's wrong with your environment~

tornadomeet avatar Mar 21 '17 08:03 tornadomeet