diff --git a/example.yaml b/example.yaml index c80a2414cab848bd6dfe8865f526911d70ada5f0..801b3a3c112813b9daf89fe7e3192ba5e6165c11 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 0a780eddd6b12f9ebb6afc028a500e7d70c0030d..eae0bd705667bf497343d53f6a3c52aafd17ef24 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