Skip to content

Commit

Permalink
Fixup
Browse files Browse the repository at this point in the history
  • Loading branch information
qubvel committed Oct 7, 2024
1 parent e2a791f commit d80dce4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion segmentation_models_pytorch/losses/_functional.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def soft_tversky_score(
"""
assert output.size() == target.size()

if dims is not None:
output_sum = torch.sum(output, dim=dims)
target_sum = torch.sum(target, dim=dims)
Expand Down

0 comments on commit d80dce4

Please sign in to comment.