浏览代码

change model dim

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

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

@@ -94,6 +94,7 @@ class DSSMLayer(nn.Layer):
             embedded_features.append(embedded)
         
         # 将所有embedding连接起来
+        paddle.static.Print(left_embedded, message="lqc debug embedded_features shape:")
         return paddle.concat(embedded_features, axis=1)
 
     def forward(self, left_features, right_features):
@@ -112,9 +113,11 @@ class DSSMLayer(nn.Layer):
     def get_vectors(self, left_features, right_features):
         """获取两个视频的16维特征向量"""
         # 处理左视频特征
+        
         left_embedded = self._process_features(left_features, self.left_embeddings)
-        paddle.static.Print(left_embedded, message="left_embedded shape:")
+        paddle.static.Print(left_embedded, message="lqc left_embedded shape:")
         left_vec = left_embedded
+        paddle.static.Print(left_vec, message=f"lqc lqc left_vec {i}:")
         for i, layer in enumerate(self._left_tower):
             left_vec = layer(left_vec)
             paddle.static.Print(left_vec, message=f"After layer {i}:")