often 5 月之前
父節點
當前提交
3bc4bccc79
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      recommend-model-produce/src/main/python/models/dssm/net.py

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