|
@@ -190,6 +190,7 @@ class DSSMLayer(nn.Layer):
|
|
|
|
|
|
|
|
|
def forward(self, left_features, right_features, is_infer=False):
|
|
|
+ paddle.static.Print(left_features, message="left_features shape and value:")
|
|
|
if is_infer:
|
|
|
left_vec = self.get_leftvectors(left_features)
|
|
|
return left_vec
|
|
@@ -219,7 +220,7 @@ class DSSMLayer(nn.Layer):
|
|
|
|
|
|
for i, layer in enumerate(self._left_tower):
|
|
|
left_vec = layer(left_vec)
|
|
|
-
|
|
|
+ left_vec = F.normalize(left_vec, p=2, axis=1)
|
|
|
return left_vec
|
|
|
|
|
|
|
|
@@ -237,9 +238,8 @@ class DSSMLayer(nn.Layer):
|
|
|
|
|
|
|
|
|
# 确保输出是L2归一化的
|
|
|
- left_vec = F.normalize(left_vec, p=2, axis=1)
|
|
|
+
|
|
|
right_vec = F.normalize(right_vec, p=2, axis=1)
|
|
|
-
|
|
|
return right_vec
|
|
|
|
|
|
|