@@ -89,6 +89,7 @@ class DSSMLayer(nn.Layer):
starts=[i],
ends=[i+1]
)
+ feature = paddle.cast(feature, dtype='int64')
embedded = embeddings[i](feature)
embedded_features.append(embedded)