update jet ptcut to be the same as the PF paper (#283)

jpata · Dec 18, 2023 · bae2907 · bae2907
1 parent f70e596
commit bae2907
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mlpf/pyg/inference.py b/mlpf/pyg/inference.py
@@ -131,7 +131,7 @@ def predict_one_batch(conv_type, model, i, batch, rank, jetdef, jet_ptcut, jet_m
 
 
 @torch.no_grad()
-def run_predictions(world_size, rank, model, loader, sample, outpath, jetdef, jet_ptcut=5.0, jet_match_dr=0.1, dir_name=""):
+def run_predictions(world_size, rank, model, loader, sample, outpath, jetdef, jet_ptcut=15.0, jet_match_dr=0.1, dir_name=""):
     """Runs inference on the given sample and stores the output as .parquet files."""
     if world_size > 1:
         conv_type = model.module.conv_type

diff --git a/mlpf/pyg/training.py b/mlpf/pyg/training.py
@@ -648,7 +648,7 @@ def run(rank, world_size, config, args, outdir, logfile):
                     sample,
                     outdir,
                     jetdef,
-                    jet_ptcut=5.0,
+                    jet_ptcut=15.0,
                     jet_match_dr=0.1,
                     dir_name=testdir_name,
                 )