dbg

ppcls/arch/backbone/base/theseus_layer.py

@@ -14,7 +14,7 @@ class Identity(nn.Layer):
 class TheseusLayer(nn.Layer):
     def __init__(self, *args, **kwargs):
         super(TheseusLayer, self).__init__()
-        self.res_dict = None
+        self.res_dict = {}
 
     # stop doesn't work when stop layer has a parallel branch.
     def stop_after(self, stop_layer_name: str):

ppcls/arch/backbone/legendary_models/vgg.py

@@ -154,7 +154,7 @@ class VGGNet(TheseusLayer):
         x = self.drop(x)
         x = self.fc3(x)
         if self.res_dict:
-            for res_key in self.res_dict:
+            for res_key in list(self.res_dict):
                 res_dict[res_key] = self.res_dict.pop(res_key)
         return x