diff --git a/mindinsight/mindconverter/graph_based_converter/mapper/impl/nn/relu_mapper.py b/mindinsight/mindconverter/graph_based_converter/mapper/impl/nn/relu_mapper.py index 3ed70996..80f02877 100644 --- a/mindinsight/mindconverter/graph_based_converter/mapper/impl/nn/relu_mapper.py +++ b/mindinsight/mindconverter/graph_based_converter/mapper/impl/nn/relu_mapper.py @@ -14,7 +14,6 @@ # ============================================================================== """Mapper module.""" from mindinsight.mindconverter.graph_based_converter.mapper.base import ONNXToMindSporeMapper -from mindinsight.mindconverter.graph_based_converter.mapper.gen_setting import Setting class ReLUMapper(ONNXToMindSporeMapper): @@ -22,12 +21,12 @@ class ReLUMapper(ONNXToMindSporeMapper): @staticmethod def _operation_name_in_ms(*args, **kwargs): - if not kwargs.get('params'): + if not kwargs.get('weights'): name = "nn.ReLU" else: - params = kwargs['params'] - max_clip = params['max'] if params.get('max') else 0 - min_clip = params['min'] if params.get('min') else 0 + weights = kwargs['weights'] + min_clip = weights[0].value if weights[0] else 0 + max_clip = weights[1].value if weights[1] else 0 if max_clip == 6 and min_clip == 0: name = "nn.ReLU6" elif max_clip == min_clip == 0: @@ -43,7 +42,3 @@ class ReLUMapper(ONNXToMindSporeMapper): @staticmethod def _convert_trained_weights(**kwargs): return dict() - - @staticmethod - def _convert_settings(**kwargs): - return Setting()