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)