dbg

ppcls/arch/__init__.py

@@ -57,7 +57,8 @@ class RecModel(nn.Layer):
         x = self.backbone(x)
         if self.neck is not None:
             x = self.neck(x)
-        y = x
         if self.head is not None:
             y = self.head(x, label)
+        else:
+            y = None
         return {"features": x, "logits": y}