often 5 月之前
父节点
当前提交
da571d8f1e
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      recommend-model-produce/src/main/python/models/dssm/net.py

+ 3 - 3
recommend-model-produce/src/main/python/models/dssm/net.py

@@ -190,6 +190,7 @@ class DSSMLayer(nn.Layer):
 
 
     def forward(self, left_features, right_features, is_infer=False):
+        paddle.static.Print(left_features, message="left_features shape and value:")
         if is_infer:
             left_vec = self.get_leftvectors(left_features)
             return left_vec
@@ -219,7 +220,7 @@ class DSSMLayer(nn.Layer):
               
         for i, layer in enumerate(self._left_tower):
             left_vec = layer(left_vec)
-        
+        left_vec = F.normalize(left_vec, p=2, axis=1)
         return left_vec
 
 
@@ -237,9 +238,8 @@ class DSSMLayer(nn.Layer):
 
             
         # 确保输出是L2归一化的
-        left_vec = F.normalize(left_vec, p=2, axis=1)
+
         right_vec = F.normalize(right_vec, p=2, axis=1)
-        
         return right_vec