From 91d8c0d469e70aab61d537ea6cd5bc82f0d96b45 Mon Sep 17 00:00:00 2001 From: SunAhong1993 Date: Wed, 11 Nov 2020 15:42:21 +0800 Subject: [PATCH] fix the bug --- .../optimizer/elimination/dygraph/transpose_elimination.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/x2paddle/optimizer/elimination/dygraph/transpose_elimination.py b/x2paddle/optimizer/elimination/dygraph/transpose_elimination.py index b22c71c..b7cbf03 100644 --- a/x2paddle/optimizer/elimination/dygraph/transpose_elimination.py +++ b/x2paddle/optimizer/elimination/dygraph/transpose_elimination.py @@ -93,7 +93,7 @@ class Dygraph_TransposeElimination(FuseBase): if _graph.layers[out].outputs[0] in _graph.outputs: can_be_optimized = False break - if not _graph.layers[out].attrs.get('keepdim', False): + if _graph.layers[out].attrs.get('keepdim', False): can_be_optimized = False break propagate_layers.append(out) @@ -206,7 +206,7 @@ class Dygraph_TransposeElimination(FuseBase): if _graph.layers[ipt].outputs[0] in _graph.outputs: can_be_optimized = False break - if not _graph.layers[ipt].attrs.get('keepdim', + if _graph.layers[ipt].attrs.get('keepdim', False): can_be_optimized = False break -- GitLab