提交 fba556a6 编写于 作者: L liuqi

Add extreme option for using winograd convolution algorithm.

上级 2f6621e4
......@@ -18,6 +18,7 @@ models:
limit_opencl_kernel_time: 0
dsp_mode: 0
obfuscate: 1
extreme: 0
capture_net:
platform: caffe
model_file_path: path/to/model.prototxt
......@@ -32,3 +33,4 @@ models:
limit_opencl_kernel_time: 1
dsp_mode: 0
obfuscate: 1
extreme: 0
......@@ -25,4 +25,5 @@ bazel-bin/lib/python/tools/converter --platform=${PLATFORM} \
--input_shape=${INPUT_SHAPE} \
${DSP_MODE_FLAG} \
--embed_model_data=${EMBED_MODEL_DATA} \
--winograd=${EXTREME} \
--obfuscate=${OBFUSCATE} || exit 1
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册