Parcourir la source

change model dim

often il y a 5 mois
Parent
commit
b9589d5ec9
1 fichiers modifiés avec 2 ajouts et 1 suppressions
  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):