diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index a574449d3bb81a73566dd2cfaae935b7c991d9c9..4fb6d388076f241b8ec8a4331df7ef4612a4b722 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -151,7 +151,7 @@ model_tests: python tools/converter.py run --config=${CONF_FILE} --target_socs=$TARGET_SOCS --round=1 --validate --model_graph_format=code --model_data_format=file || exit 1; python tools/converter.py run --config=${CONF_FILE} --target_socs=$TARGET_SOCS --example --round=1 --validate --model_graph_format=code --model_data_format=file || exit 1; python tools/converter.py benchmark --config=${CONF_FILE} --target_socs=$TARGET_SOCS --round=5 --model_graph_format=code --model_data_format=file || exit 1; - + - rm -rf mace-models quantization_tests: @@ -167,7 +167,7 @@ quantization_tests: fi - if [ -z "$TARGET_SOCS" ]; then TARGET_SOCS=random; fi - > - for CONF_FILE in mace-models/mobilenet-v1/mobilenet-v1-quantize-retrain.yml mace-models/mobilenet-v1/mobilenet-v1-quantize-retrain-for-check-only.yml mace-models/mobilenet-v1/mobilenet-v1-quantize-retrain-dsp.yml; + for CONF_FILE in mace-models/mobilenet-v1/mobilenet-v1-quantize-friendly.yml mace-models/mobilenet-v1/mobilenet-v1-quantize-retrain-for-check-only.yml mace-models/mobilenet-v1/mobilenet-v1-quantize-retrain-dsp.yml; do python tools/converter.py convert --config=${CONF_FILE} --target_socs=$TARGET_SOCS --model_graph_format=file --model_data_format=file || exit 1; python tools/converter.py run --config=${CONF_FILE} --target_socs=$TARGET_SOCS --device_yml=${DEVICE_CONF_FILE} --round=1 --validate --model_graph_format=file --model_data_format=file || exit 1; diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py index 437fe91eceb1d2099aec700134bf69e8576268a4..e654a8fbebce5eddabef24495fb3e3c16fda658e 100644 --- a/mace/python/tools/converter_tool/transformer.py +++ b/mace/python/tools/converter_tool/transformer.py @@ -1713,8 +1713,8 @@ class Transformer(base_converter.ConverterInterface): mace_check(output in self._quantize_activation_info, "%s does not have quantize activation info" % op) - op.quantize_info.append( - self._quantize_activation_info[output]) + op.quantize_info.extend([ + self._quantize_activation_info[output]]) if not self._option.quantize: return False