Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix multilabel in exact_match #1474

Merged
merged 14 commits into from Feb 6, 2023
1 change: 0 additions & 1 deletion README.md
Expand Up @@ -184,7 +184,6 @@ def metric_ddp(rank, world_size):
n_epochs = 5
# this shows iteration over multiple training epochs
for n in range(n_epochs):

# this will be replaced by a DataLoader with a DistributedSampler
n_batches = 10
for i in range(n_batches):
Expand Down
Expand Up @@ -232,7 +232,7 @@ def exact_match(
if task == "multiclass":
assert num_classes is not None
return multiclass_exact_match(preds, target, num_classes, multidim_average, ignore_index, validate_args)
if task == "multilalbe":
if task == "multilabel":
SkafteNicki marked this conversation as resolved.
Show resolved Hide resolved
assert num_labels is not None
return multilabel_exact_match(
preds, target, num_labels, threshold, multidim_average, ignore_index, validate_args
Expand Down