File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
torch/_refs/nn/functional Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -528,9 +528,10 @@ def nll_loss(
528
528
lambda : f"Expected input tensor to have 1 or more dimensions (got { input .ndim } )" ,
529
529
)
530
530
531
+ # TODO: raise except
6574
ion instead of converting value
532
+ # msg = "size_average and reduce args are deprecated, please use reduction argument."
533
+ # Convert these options for consistency with the eager mode
531
534
if size_average is not None or reduce is not None :
532
- # TODO: raise exception instead of converting value
533
- # msg = "size_average and reduce args are deprecated, please use reduction argument."
534
535
reduction = _get_string_reduction_arg (size_average = size_average , reduce = reduce )
535
536
536
537
# The expected behavior when the target and input have zero elements:
You can’t perform that action at this time.
0 commit comments