如何在TinyDNN中包含和使用NNPack

Question

我刚刚开始学习tinydnn框架和nnpack（以及一些基本的机器学习概念），我不熟悉像cmake这样的工具。 目前我已按照GitHub（https://github.com/Maratyszcza/NNPACK）在MacOS X 10.12上的说明下载并构建了NNPack。此外，我已从GitHub下载了TinyDNN。但是，我不确定如何在TinyDNN中启用NNPack。从GitHub上的教程（https://github.com/tiny-dnn/tiny-dnn）我知道我可以使用cmake选项，如：     cmake。 USE_NNPACK = ON 并设置NNPACK的include和library目录。但是，我不确定如何设置目录。

谢谢！

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);

请记住，tiny-dnn还不支持使用nnpack内核进行反向传递。仅用于推理。