diff --git a/paddleslim/nas/ofa/ofa.py b/paddleslim/nas/ofa/ofa.py index b725c93a63fe3e1e2a69467d04f18d171933fd65..eff589a4a7eb7445851e32be7b66dc37c3f66513 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( )