Browse Source

change model dim

often 5 months ago
parent
commit
b9589d5ec9
1 changed files with 2 additions and 1 deletions
  1. 2 1
      recommend-model-produce/src/main/python/models/dssm/net.py

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

@@ -91,10 +91,11 @@ class DSSMLayer(nn.Layer):
             )
             feature = paddle.cast(feature, dtype='int64')
             embedded = embeddings[i](feature)
+            paddle.static.Print(embedded, message="lqc debug _process_features embedded {i}:")
             embedded_features.append(embedded)
         
         # 将所有embedding连接起来
-        paddle.static.Print(embedded_features, message="lqc debug embedded_features shape:")
+   
         return paddle.concat(embedded_features, axis=1)
 
     def forward(self, left_features, right_features):