Browse Source

change model dim

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

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

@@ -113,11 +113,11 @@ class DSSMLayer(nn.Layer):
         """获取两个视频的16维特征向量"""
         # 处理左视频特征
         left_embedded = self._process_features(left_features, self.left_embeddings)
-        fluid.layers.Print(left_embedded, message="left_embedded shape:")
+        paddle.static.Print(left_embedded, message="left_embedded shape:")
         left_vec = left_embedded
         for layer in self._left_tower:
             left_vec = layer(left_vec)
-            fluid.layers.Print(left_vec, message=f"After layer {i}:")
+            paddle.static.Print(left_vec, message=f"After layer {i}:")
         
         # 处理右视频特征
         right_embedded = self._process_features(right_features, self.right_embeddings)