From c87b7f4c7d29a0d953219a03e22648fe291d648b Mon Sep 17 00:00:00 2001
From: ceci3 <ceci3@users.noreply.github.com>
Date: Thu, 9 Sep 2021 13:59:43 +0800
Subject: [PATCH] fix ofa when teacher model is ofa model (#885)

---
 paddleslim/nas/ofa/ofa.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddleslim/nas/ofa/ofa.py b/paddleslim/nas/ofa/ofa.py
index b725c93a..eff589a4 100644
--- a/paddleslim/nas/ofa/ofa.py
+++ b/paddleslim/nas/ofa/ofa.py
@@ -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(
         )
 
-- 
GitLab