KeyKy / mobilenet-mxnet

mobilenet-mxnet
145 stars 28 forks source link

About training speed #10

Closed sunbinbin1991 closed 6 years ago

sunbinbin1991 commented 7 years ago

I use mxnet-mobilenet for train, but for my GPU 1080Ti, the input is 224*224,my training speed is about 50 sample/second,so is something wrong for my training, I can't find the reason, looking forward your reply

sunbinbin1991 commented 6 years ago

After build a new version of mxnet, problem solved

408550969 commented 6 years ago

Hi, what's your speed after solved the problem?

sunbinbin1991 commented 6 years ago

It depends batchsize you choose, in my test my batchsize is about 128,and training samples about 955 sample/s, may this could helps

408550969 commented 6 years ago

Thanks! Excuse me,what's your GPU?

sunbinbin1991 commented 6 years ago

GPU 1080TI

408550969 commented 6 years ago

Thanks!