From ce43150ffecd0912c64034c91259072b6d76d4e8 Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Sat, 5 Jun 2021 17:45:31 +0800
Subject: [PATCH] dbg

---
 ppcls/arch/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppcls/arch/__init__.py b/ppcls/arch/__init__.py
index 34b3ef1b..4c4521be 100644
--- a/ppcls/arch/__init__.py
+++ b/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}
-- 
GitLab