From 911dbb26ce6c49c88940b0e11fa704c16b4dda6e Mon Sep 17 00:00:00 2001 From: liuqi Date: Thu, 17 May 2018 10:24:11 +0800 Subject: [PATCH] Fix fc shape converter bug. --- mace/python/tools/converter_tool/base_converter.py | 2 +- mace/python/tools/converter_tool/transformer.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mace/python/tools/converter_tool/base_converter.py b/mace/python/tools/converter_tool/base_converter.py index e6ba45f4..277727d4 100644 --- a/mace/python/tools/converter_tool/base_converter.py +++ b/mace/python/tools/converter_tool/base_converter.py @@ -210,9 +210,9 @@ class ConverterOption(object): TransformerRule.FOLD_BIASADD, TransformerRule.FOLD_ACTIVATION, TransformerRule.TRANSPOSE_FILTERS, - TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSPOSE_DATA_FORMAT, TransformerRule.TRANSFORM_GLOBAL_CONV_TO_FC, + TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSFORM_BUFFER_IMAGE, TransformerRule.ADD_DEVICE_AND_DATA_TYPE, TransformerRule.SORT_BY_EXECUTION, diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py index 14b76893..a1776b22 100644 --- a/mace/python/tools/converter_tool/transformer.py +++ b/mace/python/tools/converter_tool/transformer.py @@ -64,9 +64,9 @@ class Transformer(base_converter.ConverterInterface): TransformerRule.FOLD_BIASADD, TransformerRule.FOLD_ACTIVATION, TransformerRule.TRANSPOSE_FILTERS, - TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSPOSE_DATA_FORMAT, TransformerRule.TRANSFORM_GLOBAL_CONV_TO_FC, + TransformerRule.RESHAPE_FC_WEIGHT, TransformerRule.TRANSFORM_BUFFER_IMAGE, TransformerRule.ADD_DEVICE_AND_DATA_TYPE, TransformerRule.SORT_BY_EXECUTION, @@ -88,10 +88,10 @@ class Transformer(base_converter.ConverterInterface): TransformerRule.FOLD_BIASADD: self.fold_biasadd, TransformerRule.FOLD_ACTIVATION: self.fold_activation, TransformerRule.TRANSPOSE_FILTERS: self.transpose_filters, - TransformerRule.RESHAPE_FC_WEIGHT: self.reshape_fc_weight, TransformerRule.TRANSPOSE_DATA_FORMAT: self.transpose_data_format, TransformerRule.TRANSFORM_GLOBAL_CONV_TO_FC: self.transform_global_conv_to_fc, + TransformerRule.RESHAPE_FC_WEIGHT: self.reshape_fc_weight, TransformerRule.TRANSFORM_BUFFER_IMAGE: self.transform_buffer_image, TransformerRule.ADD_DEVICE_AND_DATA_TYPE: -- GitLab