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