often hai 5 meses
pai
achega
3bc4bccc79
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  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)