ShuffleNet icon indicating copy to clipboard operation
ShuffleNet copied to clipboard

about the train.prototxt and deploy.prototxt

Open abcdvzz opened this issue 8 years ago • 1 comments

When I use your delpoy.prototxt, do i need to add a fc layer after the last conv layer? or i just need a softmax layer

abcdvzz avatar Dec 24 '17 02:12 abcdvzz

@abcdvzz The deploy is for imagenet 1000 classification, so I have fc1000 layer, but if you do other classification task, you may change to your own fc layer. Besides, you also need an input data layers and softmax layer for training.

farmingyard avatar May 22 '18 01:05 farmingyard