diff --git a/paddleslim/nas/ofa/convert_super.py b/paddleslim/nas/ofa/convert_super.py index 07619109cedfd15d0f77bfdd315e1d6e61c8ba83..a73f9e230ca38181f8cb2a234932877a7e52f2bf 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)