未验证 提交 40282d67 编写于 作者: C ceci3 提交者: GitHub

Fix ofa convert bug(#640)

上级 eae2a994
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册