diff --git a/example.yaml b/example.yaml index 801b3a3c112813b9daf89fe7e3192ba5e6165c11..f34c12984c6d466129d696d8a2530bfced44da52 100644 --- a/example.yaml +++ b/example.yaml @@ -18,7 +18,7 @@ models: limit_opencl_kernel_time: 0 dsp_mode: 0 obfuscate: 1 - extreme: 0 + fast_conv: 0 capture_net: platform: caffe model_file_path: path/to/model.prototxt @@ -33,4 +33,4 @@ models: limit_opencl_kernel_time: 1 dsp_mode: 0 obfuscate: 1 - extreme: 0 + fast_conv: 0 diff --git a/generate_model_code.sh b/generate_model_code.sh index eae0bd705667bf497343d53f6a3c52aafd17ef24..23cc203292c6b6afbf61e649212b9d016a075a15 100644 --- a/generate_model_code.sh +++ b/generate_model_code.sh @@ -25,5 +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} \ + --winograd=${FAST_CONV} \ --obfuscate=${OBFUSCATE} || exit 1