From 79b3e3b022e553fc3c8762600d116015422e18a0 Mon Sep 17 00:00:00 2001 From: Farouk Date: Wed, 27 Nov 2024 13:28:07 +0100 Subject: [PATCH] up --- mlpf/model/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlpf/model/inference.py b/mlpf/model/inference.py index e4d2d6a4d..a07860246 100644 --- a/mlpf/model/inference.py +++ b/mlpf/model/inference.py @@ -147,7 +147,7 @@ def predict_one_batch_args(args): @torch.no_grad() def run_predictions( - world_size, rank, model, loader, sample, outpath, jetdef, jet_ptcut=15.0, jet_etacut=2.5, jet_match_dr=0.1, dir_name="" + world_size, rank, model, loader, sample, outpath, jetdef, jet_ptcut=15.0, jet_etacut=0.5, jet_match_dr=0.1, dir_name="" ): """Runs inference on the given sample and stores the output as .parquet files.""" if world_size > 1: