Skip to content

Commit 1a41ce5

Browse files
committed
FIX Fixes onnx
1 parent 937ab19 commit 1a41ce5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch/onnx/symbolic_opset12.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ def cross_entropy_loss(g, self, target, weight, reduction, ignore_index, label_s
7373
reduction_vals = ["none", "mean", "sum"]
7474
reduction = reduction_vals[reduction]
7575

76+
label_smoothing = sym_help._maybe_get_const(label_smoothing, "f")
7677
if label_smoothing > 0.0:
7778
raise RuntimeError("Unsupported: ONNX does not support label_smoothing")
7879

0 commit comments

Comments
 (0)