adjust soft_test. (#1613)

PaddlePaddle · Dec 28, 2022 · 3ee00dd · 3ee00dd
1 parent f68ec4b
commit 3ee00dd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_soft_label_loss.py b/tests/test_soft_label_loss.py
@@ -54,9 +54,12 @@ def test_soft_label_loss(self):
         for block in paddle.static.default_main_program().blocks:
             for op in block.ops:
                 loss_ops.append(op.type)
+        print(f"ret: {set(loss_ops).difference(set(merged_ops))}")
         self.assertTrue(set(merged_ops).difference(set(loss_ops)) == set())
-        self.assertTrue({'cross_entropy', 'softmax', 'reduce_mean'}.issubset(
-            set(loss_ops).difference(set(merged_ops))))
+
+        self.assertTrue({
+            'softmax_with_cross_entropy', 'softmax', 'reduce_mean'
+        }.issubset(set(loss_ops).difference(set(merged_ops))))
 
 
 if __name__ == '__main__':