We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f68ec4b commit 3ee00ddCopy full SHA for 3ee00dd
tests/test_soft_label_loss.py
@@ -54,9 +54,12 @@ def test_soft_label_loss(self):
54
for block in paddle.static.default_main_program().blocks:
55
for op in block.ops:
56
loss_ops.append(op.type)
57
+ print(f"ret: {set(loss_ops).difference(set(merged_ops))}")
58
self.assertTrue(set(merged_ops).difference(set(loss_ops)) == set())
- self.assertTrue({'cross_entropy', 'softmax', 'reduce_mean'}.issubset(
59
- set(loss_ops).difference(set(merged_ops))))
+
60
+ self.assertTrue({
61
+ 'softmax_with_cross_entropy', 'softmax', 'reduce_mean'
62
+ }.issubset(set(loss_ops).difference(set(merged_ops))))
63
64
65
if __name__ == '__main__':
0 commit comments