From 40282d674d83d164a7e5126a8c5a3f6b211934a3 Mon Sep 17 00:00:00 2001 From: ceci3 Date: Tue, 2 Feb 2021 10:10:28 +0800 Subject: [PATCH] Fix ofa convert bug(#640) --- paddleslim/nas/ofa/convert_super.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddleslim/nas/ofa/convert_super.py b/paddleslim/nas/ofa/convert_super.py index 07619109..a73f9e23 100644 --- a/paddleslim/nas/ofa/convert_super.py +++ b/paddleslim/nas/ofa/convert_super.py @@ -431,7 +431,7 @@ class Convert: new_attr_name = ['act', 'dtype'] else: new_attr_name = ['weight_attr', 'bias_attr'] - self._change_name(layer, pd_ver) + self._change_name(layer, pd_ver) if pd_ver != 185 else None in_nc, out_nc = layer._parameters['weight'].shape new_attr_dict = dict.fromkeys(new_attr_name, None) -- GitLab