From 7dc472e333e981c0335134f508a840a0cebfb64a Mon Sep 17 00:00:00 2001 From: Siddharth Singh Date: Thu, 31 Aug 2023 03:41:21 -0400 Subject: [PATCH] Correct Confusion Matrix All-reduce to display correct IoUs on the master process --- references/segmentation/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/references/segmentation/utils.py b/references/segmentation/utils.py index cb200f23d76..92db1899851 100644 --- a/references/segmentation/utils.py +++ b/references/segmentation/utils.py @@ -88,7 +88,7 @@ def compute(self): return acc_global, acc, iu def reduce_from_all_processes(self): - reduce_across_processes(self.mat) + self.mat = reduce_across_processes(self.mat).to(torch.int64) def __str__(self): acc_global, acc, iu = self.compute()