From a86edf5ac4edd26777a69d70bb4bc4f5fb6776c8 Mon Sep 17 00:00:00 2001 From: ceci3 <ceci3@users.noreply.github.com> Date: Tue, 2 Feb 2021 10:11:08 +0800 Subject: [PATCH] fix ofa convert bug(#641) --- 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