Skip to content

Commit 3ee00dd

Browse files
authored
adjust soft_test. (#1613)
1 parent f68ec4b commit 3ee00dd

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/test_soft_label_loss.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -54,9 +54,12 @@ def test_soft_label_loss(self):
5454
for block in paddle.static.default_main_program().blocks:
5555
for op in block.ops:
5656
loss_ops.append(op.type)
57+
print(f"ret: {set(loss_ops).difference(set(merged_ops))}")
5758
self.assertTrue(set(merged_ops).difference(set(loss_ops)) == set())
58-
self.assertTrue({'cross_entropy', 'softmax', 'reduce_mean'}.issubset(
59-
set(loss_ops).difference(set(merged_ops))))
59+
60+
self.assertTrue({
61+
'softmax_with_cross_entropy', 'softmax', 'reduce_mean'
62+
}.issubset(set(loss_ops).difference(set(merged_ops))))
6063

6164

6265
if __name__ == '__main__':

0 commit comments

Comments
 (0)