|
@@ -115,7 +115,7 @@ class DSSMLayer(nn.Layer):
|
|
|
left_embedded = self._process_features(left_features, self.left_embeddings)
|
|
|
paddle.static.Print(left_embedded, message="left_embedded shape:")
|
|
|
left_vec = left_embedded
|
|
|
- for layer in self._left_tower:
|
|
|
+ for i, layer in enumerate(self._left_tower):
|
|
|
left_vec = layer(left_vec)
|
|
|
paddle.static.Print(left_vec, message=f"After layer {i}:")
|
|
|
|