add custom op & kd loss

alibaba · Aug 21, 2024 · 262e2ca · 262e2ca
1 parent 645b0dd
commit 262e2ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easy_rec/python/loss/listwise_loss.py b/easy_rec/python/loss/listwise_loss.py
@@ -44,7 +44,7 @@ def listwise_rank_loss(labels,
   """
   loss_name = name if name else 'listwise_rank_loss'
   logging.info('[{}] temperature: {}'.format(loss_name, temperature))
-
+  labels = tf.to_float(labels)
   if temperature != 1.0:
     logits /= temperature
     if label_is_logits: