Function reduce_mean

tools/test_reconstruction_hf.py:417–422

(values)

    # Aggregate metrics
    if use_ddp:
        def reduce_mean(values):
            if not values:
                return None
            t = torch.tensor(np.mean(values), device=device)
            dist.all_reduce(t, op=dist.ReduceOp.SUM)
            return (t / world_size).item()

        avg_lpips = reduce_mean(lpips_values)
        avg_psnr = reduce_mean(psnr_values)