From fba556a6a094a71008d0a67b71998be8657a47b8 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Tue, 6 Mar 2018 10:49:32 +0800
Subject: [PATCH] Add extreme option for using winograd convolution algorithm.

---
 example.yaml           | 2 ++
 generate_model_code.sh | 1 +
 2 files changed, 3 insertions(+)

diff --git a/example.yaml b/example.yaml
index c80a2414..801b3a3c 100644
--- a/example.yaml
+++ b/example.yaml
@@ -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
diff --git a/generate_model_code.sh b/generate_model_code.sh
index 0a780edd..eae0bd70 100644
--- a/generate_model_code.sh
+++ b/generate_model_code.sh
@@ -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
-- 
GitLab