search

Maratyszcza / caffe-nnpack

Caffe with NNPACK integration
Other
59 stars 28 forks source link

Request for a Train_val prototxt example #5

Closed anijain2305 closed 8 years ago

anijain2305 commented 8 years ago

Hi,

I am finally able to get the caffe code compile. However, I am not able to figure out how to instantiate a NNPack layer. In the train_val.prototxt files, a convolution layer can be instantiated using CONVOLUTION but there seems to be no easy name for nnpack. Am I missing something? Let me know how can I use the nnconvolution.

Maratyszcza commented 8 years ago

You need to specify: Engine: NNPACK inside conv_params. Also see details in Maratyszcza/NNPACK#9