often 5 ヶ月 前
コミット
3bc4bccc79

+ 1 - 0
recommend-model-produce/src/main/python/models/dssm/net.py

@@ -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)