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