|
@@ -138,6 +138,8 @@ class DSSMLayer(nn.Layer):
|
|
starts=[i],
|
|
starts=[i],
|
|
ends=[i+1]
|
|
ends=[i+1]
|
|
)
|
|
)
|
|
|
|
+ paddle.static.Print(i, message="debug debug:")
|
|
|
|
+ # paddle.static.Print(embeddings[i], message="debug debug embeddings i:")
|
|
feature = paddle.cast(feature, dtype='int64')
|
|
feature = paddle.cast(feature, dtype='int64')
|
|
embedded = embeddings[i](feature)
|
|
embedded = embeddings[i](feature)
|
|
embedded_features.append(embedded)
|
|
embedded_features.append(embedded)
|