From 7cf054de2555bb8e8345b9308f5ceb5d0127c575 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Fri, 22 Oct 2021 12:50:23 +0100 Subject: [PATCH] Fix quantization error. --- references/classification/utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/references/classification/utils.py b/references/classification/utils.py index 1a4adc7f60f..e7318f40f04 100644 --- a/references/classification/utils.py +++ b/references/classification/utils.py @@ -33,7 +33,6 @@ def synchronize_between_processes(self): Warning: does not synchronize the deque! """ t = reduce_across_processes([self.count, self.total]) - t = t.tolist() self.count = int(t[0]) self.total = t[1] @@ -407,4 +406,4 @@ def reduce_across_processes(val): t = torch.tensor(val, device="cuda") dist.barrier() dist.all_reduce(t) - return t + return t.tolist()