often 5 kuukautta sitten
vanhempi
commit
3bc4bccc79
1 muutettua tiedostoa jossa 1 lisäystä ja 0 poistoa
  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)