|
|
@@ -143,7 +143,7 @@ class DSSMLayer(nn.Layer):
|
|
|
|
|
|
def _process_features(self, features, embeddings):
|
|
|
embedded_features = []
|
|
|
- paddle.static.Print(features, message="Input features shape:")
|
|
|
+
|
|
|
|
|
|
# 1. 处理视频静态特征 (使用Embedding)
|
|
|
for i in range(self.feature_num):
|
|
|
@@ -250,16 +250,11 @@ class DSSMLayer(nn.Layer):
|
|
|
def get_vectors(self, left_features, right_features):
|
|
|
"""获取两个视频的16维特征向量"""
|
|
|
# 处理左视频特征
|
|
|
-
|
|
|
+ paddle.static.Print(left_features, message="left_features shape:")
|
|
|
left_embedded = self._process_features(left_features, self.left_embeddings)
|
|
|
|
|
|
# left_vec = left_embedded
|
|
|
left_vec = paddle.reshape(left_embedded, [-1, self.feature_num * self.embedding_dim])
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
|
|
|
for i, layer in enumerate(self._left_tower):
|
|
|
left_vec = layer(left_vec)
|
often