[fbsync] Fix IoUs reported in segmentation references (#7916)

Summary: Co-authored-by: Nicolas Hug <[email protected]>

Reviewed By: matteobettini

Differential Revision: D49600769

fbshipit-source-id: 968021fd29ea9699cb4924e23709f00b5e85ce7b

Author: NicolasHug

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()