We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 937ab19 commit 1a41ce5Copy full SHA for 1a41ce5
torch/onnx/symbolic_opset12.py
@@ -73,6 +73,7 @@ def cross_entropy_loss(g, self, target, weight, reduction, ignore_index, label_s
73
reduction_vals = ["none", "mean", "sum"]
74
reduction = reduction_vals[reduction]
75
76
+ label_smoothing = sym_help._maybe_get_const(label_smoothing, "f")
77
if label_smoothing > 0.0:
78
raise RuntimeError("Unsupported: ONNX does not support label_smoothing")
79
0 commit comments