forresti / SqueezeNet

SqueezeNet: AlexNet-level accuracy with 50x fewer parameters
BSD 2-Clause "Simplified" License
2.17k stars 723 forks source link

SqueezeNet speed slower than Alexnet #29

Closed PearlDzzz closed 7 years ago

PearlDzzz commented 7 years ago

@macd @forresti @antingshen @samster25 @terrychenism Hi, I use command" caffe.exe time --model=SqueezeNet_v1.1_deploy.prototxt -gpu 0 -iterations 100 " to test the time. AlexNet:11ms, SqueezeNet:30ms. even use cudnn v4, the time of SqueezeNet is still twice or even three times than AlexNet. Do you have any advice?

chibai commented 7 years ago

The contribution of squeeze net is the model size not the computational complexity~~ Actually, computational complexity of Alexnet and SqueezeNet are similar~

maydaygmail commented 7 years ago

Interesting issue! @forresti would you please explain why much less parameters but use much more time?

austingg commented 7 years ago

@maydaygmail the most majority of model parameter is in FC layers. and the most computation cost at conv layers. Squeezenet use more Conv layers, it's much deeper than alexnet

maydaygmail commented 7 years ago

Thanks @austingg. I think there is a implementation problem. In fire module, expand layer has 2 different convolution, and the 2 different convolution are calculating serially not parallel