fix teacher_student_sigmoid_loss dtype check, test=develop (#24586)

c417f991 · Bai Yifan · GitHub · 7fa9f16c · c417f991
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

python/paddle/fluid/layers/loss.py python/paddle/fluid/layers/loss.py +4 -2

未找到文件。
--- a/python/paddle/fluid/layers/loss.py
+++ b/python/paddle/fluid/layers/loss.py
@@ -1536,9 +1536,11 @@ def teacher_student_sigmoid_loss(input,
          cost = fluid.layers.teacher_student_sigmoid_loss(input=similarity, label=label)

    """
-    check_variable_and_dtype(input, "input", ['float32', 'float64'],
+    check_variable_and_dtype(input, "input",
+                             ['float32', 'float64', 'int32', 'int64'],
                             'teacher_student_sigmoid_loss')
-    check_variable_and_dtype(label, "label", ['float32', 'float64'],
+    check_variable_and_dtype(label, "label",
+                             ['float32', 'float64', 'int32', 'int64'],
                             'teacher_student_sigmoid_loss')

    helper = LayerHelper('teacher_student_sigmoid_loss', **locals())