Skip to content

Fix Recall/Precision multi_label, not averaged for DDP  #1283

@vfdev-5

Description

@vfdev-5

🚀 Feature

Idea is to fix that

if idist.get_world_size() > 1:
if (not average) and is_multilabel:
warnings.warn(
"Precision/Recall metrics do not work in distributed setting when average=False "
"and is_multilabel=True. Results are not reduced across computing devices. Computed result "
"corresponds to the local rank's (single process) result.",
RuntimeWarning,
)

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions