TinyDNNにNNPackを組み込み使用するにはどうすればいいですか

Question

最近tinydnnフレームワークとnnpack（そしていくつかの基本的な機械学習の概念を実際に学び始めたばかりです）、そして私はcmakeのようなツールに慣れていません。 現在、私はMacOS X 10.12のGitHub（https://github.com/Maratyszcza/NNPACK）の指示に従ってNNPackをダウンロードしてビルドしました。またGitHubからTinyDNNをダウンロードしました。しかし、私はTinyDNNでどのようにNNPackを有効にできるかわかりません。 GitHub（https://github.com/tiny-dnn/tiny-dnn）のチュートリアルから、私はおそらくcmakeのようなオプションを使用できることを知っています： cmake。 USE_NNPACK = ON とし、NNPACKのインクルードおよびライブラリディレクトリを設定します。しかし、私はどのようにディレクトリを設定できるか分かりません。

ありがとうございます！

Answer 1

これをコードに追加することで有効にすることができます。

core::backend_t backend_type = parse_backend_name("nnpack");

その後、層への引数としてbackend_typeを渡す：

conv(in_width,in_height,window_size,in_channels,out_channels,pad_type, has_bias,w_stride,h_stride,backend_type);

小さな-DNNはまだnnpackカーネルとの下位パスをサポートしていませんのでご注意ください。推論のためだけ。