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

fix ofa when teacher model is ofa model (#885)

上级 358b51af
......@@ -83,6 +83,8 @@ class OFABase(Layer):
def __init__(self, model):
super(OFABase, self).__init__()
self.model = model
self._skip_layers = None
self._cannot_changed_layer = []
self._ofa_layers, self._elastic_task, self._key2name, self._layers = self.get_layers(
)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册