add cms model v2.4.0

cms/v2.4.0/pyg-cms_20250410_144744_012331/.gitattributes

@@ -0,0 +1,6 @@
+plots_checkpoint*/** filter=lfs diff=lfs merge=lfs -text
+preds_checkpoint*/** filter=lfs diff=lfs merge=lfs -text
+runs/** filter=lfs diff=lfs merge=lfs -text
+checkpoints/** filter=lfs diff=lfs merge=lfs -text
+test.log filter=lfs diff=lfs merge=lfs -text
+train.log filter=lfs diff=lfs merge=lfs -text

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch1.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9a8217b83a4e302c64d7c9b0969163622a264d69060936b298ae7b738d38acf5
+size 3748252

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch10.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9cfbca6c3cb3f9a766fa7312101efaa37b3d4d72fad8357db914a9384a2b96e2
+size 3778159

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch2.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fc0296d82189c29404c5bec9ae8d15dbe0dbfaec8896d9b717b15a38b4cbed5a
+size 3754811

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch3.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9636bedd3a4fb2067a85d2377d62782508bcb54afdce0bed172c2dfaa1c1ffa6
+size 3760948

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch4.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b0bb69f5a5da9d5239a9598a5dc61bded17d47ff00be70cef24a19907ee5f71a
+size 3762477

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch5.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:167df7cda3f9b6c4ad598c8a9b5781fd9e6fa09f296cf03216691c821db41a51
+size 3764449

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch6.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9278cd0bac3bfac526c8325b9a358d8bba74b4561e3a4a4eef505a6ceb53a024
+size 3769581

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch7.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b475cc5af7374084fd748ee372a1b28edce5da608892d528973e5658897b948
+size 3773624

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch8.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:760c5043a97c96cc575826b9c2475ba3633dc5383f1a78db708a257860ac1069
+size 3777078

cms/v2.4.0/pyg-cms_20250410_144744_012331/batch0_epoch9.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62b1d969023a03c1999c230e3bcc9040c2946c43f3e76db52ae9a4b2e45d2a47
+size 3777227

cms/v2.4.0/pyg-cms_20250410_144744_012331/checkpoints/checkpoint-10-3.539432.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d809cc077a73bd8bce8088e070b5dc7fea92546a89f98ed5fee7f332285549ed
+size 61982662

cms/v2.4.0/pyg-cms_20250410_144744_012331/checkpoints/mlpf_5M_attn2x3x256_bm5_relu_checkpoint10_pudisc_1xa100_fp32_fused.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23715e562c08201354170d9a38c014e3c06b8393509e1bb89407a9c58d5fbc3f
+size 20620703

cms/v2.4.0/pyg-cms_20250410_144744_012331/checkpoints/mlpf_5M_attn2x3x256_bm5_relu_checkpoint10_pudisc_1xa100_fp32_unfused.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:67693d0281b4f708e17fe7da9b9dcfc861b294e1d5e18ddac7bfedf4ba26cdce
+size 20685254

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_1.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.31193939394873893, "Regression_eta": 0.001314852967108619, "Regression_sin_phi": 0.000651800558687316, "Regression_cos_phi": 0.000656838548656735, "Regression_energy": 0.3584940971956668, "Classification_binary": 2.966919501550562, "Classification": 0.045426947592373756, "ispu": 0.07793570920616671, "MET": 589.3520502692821, "Sliced_Wasserstein_Loss": 175.96155297937273, "Total": 3.763300660931521}, "valid": {"Regression_pt": 0.32145090859321224, "Regression_eta": 0.0013373081639167866, "Regression_sin_phi": 0.000660682239638771, "Regression_cos_phi": 0.0006666128204011641, "Regression_energy": 0.36783735629870995, "Classification_binary": 3.068020807891289, "Classification": 0.04703030307892677, "ispu": 0.16466039164672935, "MET": 34.20720463174723, "Sliced_Wasserstein_Loss": 175.32290117014898, "Total": 3.9716819642738193}, "epoch_train_time": 56320.43389034271, "epoch_valid_time": 2072.5102972984314, "epoch_total_time": 58392.944189071655}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_10.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.2815292814990983, "Regression_eta": 0.0012761775570004096, "Regression_sin_phi": 0.0006380058848746152, "Regression_cos_phi": 0.0006427805878243366, "Regression_energy": 0.3271001583250481, "Classification_binary": 2.809757374727696, "Classification": 0.04403375405064181, "ispu": 0.07308357382252299, "MET": 539.3180291954595, "Sliced_Wasserstein_Loss": 169.48686154977216, "Total": 3.5380578382925814}, "valid": {"Regression_pt": 0.2778594067447321, "Regression_eta": 0.0012802250906145138, "Regression_sin_phi": 0.0006426413481629947, "Regression_cos_phi": 0.0006476986150688427, "Regression_energy": 0.3240199117531141, "Classification_binary": 2.81790059423363, "Classification": 0.04403388171201595, "ispu": 0.07305401214383452, "MET": 25.753635899371627, "Sliced_Wasserstein_Loss": 169.3599817927482, "Total": 3.539431814397051}, "epoch_train_time": 58375.90614724159, "epoch_valid_time": 2147.5064384937286, "epoch_total_time": 60523.41259098053}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_2.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.29531431947367254, "Regression_eta": 0.001292414016521669, "Regression_sin_phi": 0.0006445869310261045, "Regression_cos_phi": 0.0006495042320928711, "Regression_energy": 0.3412666882742015, "Classification_binary": 2.8815920456311663, "Classification": 0.04454768790383064, "ispu": 0.074383442242196, "MET": 742.5657563952958, "Sliced_Wasserstein_Loss": 172.41185482697114, "Total": 3.639690553786109}, "valid": {"Regression_pt": 0.31112914364161687, "Regression_eta": 0.0013265160188493455, "Regression_sin_phi": 0.0006560527148381123, "Regression_cos_phi": 0.0006617011975645979, "Regression_energy": 0.35743495018679083, "Classification_binary": 2.9793676588138642, "Classification": 0.04621149390015432, "ispu": 0.1442385761464185, "MET": 32.778718054044454, "Sliced_Wasserstein_Loss": 174.01526522636942, "Total": 3.841024463219131}, "epoch_train_time": 57318.1357152462, "epoch_valid_time": 2137.9930062294006, "epoch_total_time": 59456.12872552872}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_3.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.29116541837409365, "Regression_eta": 0.0012871889324018503, "Regression_sin_phi": 0.0006426352039996194, "Regression_cos_phi": 0.0006475021364601015, "Regression_energy": 0.336880677374629, "Classification_binary": 2.8617431297907667, "Classification": 0.0444020384581225, "ispu": 0.07386880488488591, "MET": 649.6535179369265, "Sliced_Wasserstein_Loss": 171.61336862906668, "Total": 3.6105780498696145}, "valid": {"Regression_pt": 0.30589774622810134, "Regression_eta": 0.0013199406486264073, "Regression_sin_phi": 0.0006545304611520806, "Regression_cos_phi": 0.0006603525394121474, "Regression_energy": 0.3499668935403113, "Classification_binary": 2.9469439710682326, "Classification": 0.0448665250021857, "ispu": 0.12740718959202987, "MET": 31.466494215866955, "Sliced_Wasserstein_Loss": 172.72118480360592, "Total": 3.777734487754513}, "epoch_train_time": 58016.60260128975, "epoch_valid_time": 2127.8043365478516, "epoch_total_time": 60144.40693926811}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_4.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.2888538002020541, "Regression_eta": 0.0012843741630228366, "Regression_sin_phi": 0.0006414311759701649, "Regression_cos_phi": 0.0006462676915144451, "Regression_energy": 0.33450271780025015, "Classification_binary": 2.8507054006912766, "Classification": 0.044305412648919296, "ispu": 0.07361666611671276, "MET": 552.8375253184387, "Sliced_Wasserstein_Loss": 171.16262270653803, "Total": 3.594585493058862}, "valid": {"Regression_pt": 0.3038105517754846, "Regression_eta": 0.0013145139988847711, "Regression_sin_phi": 0.0006524558695912094, "Regression_cos_phi": 0.0006574665507232248, "Regression_energy": 0.3466214020902258, "Classification_binary": 2.917313472811244, "Classification": 0.04469832020494954, "ispu": 0.1179018648902569, "MET": 30.07300386349002, "Sliced_Wasserstein_Loss": 170.9832582099163, "Total": 3.732955680884606}, "epoch_train_time": 58046.48045158386, "epoch_valid_time": 2144.4770090579987, "epoch_total_time": 60190.95746707916}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_5.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.28720646338725553, "Regression_eta": 0.0012823145462911108, "Regression_sin_phi": 0.0006406055524477163, "Regression_cos_phi": 0.0006454214386973447, "Regression_energy": 0.33282672042029227, "Classification_binary": 2.84201880163913, "Classification": 0.044238107542304304, "ispu": 0.07346521035274352, "MET": 498.4589514404064, "Sliced_Wasserstein_Loss": 170.8270402159191, "Total": 3.5824048877847323}, "valid": {"Regression_pt": 0.2983217271354664, "Regression_eta": 0.0013071165168057999, "Regression_sin_phi": 0.0006510645219296469, "Regression_cos_phi": 0.0006559448052456348, "Regression_energy": 0.34212037203438284, "Classification_binary": 2.9003263982947356, "Classification": 0.044614399406148, "ispu": 0.09634601003411085, "MET": 28.48349412705109, "Sliced_Wasserstein_Loss": 171.6278171281391, "Total": 3.6843428736372315}, "epoch_train_time": 58223.876123189926, "epoch_valid_time": 2139.1980402469635, "epoch_total_time": 60363.074172735214}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_6.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.28577732849456133, "Regression_eta": 0.001280621438341738, "Regression_sin_phi": 0.0006398914325215423, "Regression_cos_phi": 0.0006446736644109177, "Regression_energy": 0.33138605962446066, "Classification_binary": 2.8343421646679183, "Classification": 0.04418446776231876, "ispu": 0.07334931527654766, "MET": 513.2134100760602, "Sliced_Wasserstein_Loss": 170.4942011215359, "Total": 3.571605848960248}, "valid": {"Regression_pt": 0.29247729210843304, "Regression_eta": 0.0012981362195348147, "Regression_sin_phi": 0.0006486571385407978, "Regression_cos_phi": 0.0006531915013107279, "Regression_energy": 0.3367970146421276, "Classification_binary": 2.873318743477585, "Classification": 0.044439052586276284, "ispu": 0.08468256937524979, "MET": 27.716111197459867, "Sliced_Wasserstein_Loss": 171.91110642360727, "Total": 3.634301379976464}, "epoch_train_time": 58433.02326631546, "epoch_valid_time": 2152.5303637981415, "epoch_total_time": 60585.55363559723}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_7.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.2844534686569447, "Regression_eta": 0.001279091708382855, "Regression_sin_phi": 0.0006392602062209578, "Regression_cos_phi": 0.0006440460520013671, "Regression_energy": 0.33004327263486927, "Classification_binary": 2.826467852675675, "Classification": 0.044133362524578314, "ispu": 0.07325379772726151, "MET": 617.5218349176117, "Sliced_Wasserstein_Loss": 170.19151964908212, "Total": 3.5609890658883843}, "valid": {"Regression_pt": 0.28611566080389456, "Regression_eta": 0.001292049152923048, "Regression_sin_phi": 0.0006467961639283005, "Regression_cos_phi": 0.0006514093837081177, "Regression_energy": 0.3310176736627662, "Classification_binary": 2.8457454079423585, "Classification": 0.044295483023854, "ispu": 0.07527837137728423, "MET": 27.50118791216111, "Sliced_Wasserstein_Loss": 169.95236139174457, "Total": 3.5850342496169816}, "epoch_train_time": 58013.0616543293, "epoch_valid_time": 2157.5400128364563, "epoch_total_time": 60170.601675987244}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_8.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.28323021954571137, "Regression_eta": 0.00127776390458549, "Regression_sin_phi": 0.0006387178214082896, "Regression_cos_phi": 0.0006434939548565453, "Regression_energy": 0.3288143569424948, "Classification_binary": 2.8192861742204633, "Classification": 0.04408948025038425, "ispu": 0.07317176240079243, "MET": 554.0395670762847, "Sliced_Wasserstein_Loss": 169.8901013230966, "Total": 3.5511728512324314}, "valid": {"Regression_pt": 0.2824250104428026, "Regression_eta": 0.001285893816977265, "Regression_sin_phi": 0.0006450353910802464, "Regression_cos_phi": 0.0006496463662727791, "Regression_energy": 0.3279947468539756, "Classification_binary": 2.832234815263006, "Classification": 0.04417625325773808, "ispu": 0.07357135478537091, "MET": 26.92840886382308, "Sliced_Wasserstein_Loss": 169.80439416479783, "Total": 3.5629822423784887}, "epoch_train_time": 58406.10660195351, "epoch_valid_time": 2163.662495136261, "epoch_total_time": 60569.76910471916}

cms/v2.4.0/pyg-cms_20250410_144744_012331/history/epoch_9.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.282215824425108, "Regression_eta": 0.0012767792698575506, "Regression_sin_phi": 0.0006382791100946558, "Regression_cos_phi": 0.0006430723342359259, "Regression_energy": 0.3277871638291101, "Classification_binary": 2.8134956221613128, "Classification": 0.044054791714980376, "ispu": 0.07311441235934543, "MET": 530.0575471767936, "Sliced_Wasserstein_Loss": 169.66216259556862, "Total": 3.5432815780449354}, "valid": {"Regression_pt": 0.27936428918861156, "Regression_eta": 0.0012825823307704377, "Regression_sin_phi": 0.0006435997445828395, "Regression_cos_phi": 0.0006481435092089429, "Regression_energy": 0.3254924206416391, "Classification_binary": 2.8231122812909386, "Classification": 0.0440825152270633, "ispu": 0.0730919529532662, "MET": 26.20586961831383, "Sliced_Wasserstein_Loss": 170.4326664742323, "Total": 3.547736791415947}, "epoch_train_time": 58554.96111559868, "epoch_valid_time": 2158.6684477329254, "epoch_total_time": 60713.62957859039}

cms/v2.4.0/pyg-cms_20250410_144744_012331/hyperparameters.json

@@ -0,0 +1 @@
+{"num_mlpf_params": 5133603, "train": true, "test": null, "make_plots": null, "comet": true, "save_attention": false, "dataset": "cms", "sort_data": true, "data_dir": "/scratch/persistent/joosep/tensorflow_datasets", "gpus": 1, "gpu_batch_multiplier": 5, "load": null, "finetune": null, "num_epochs": 10, "patience": 20, "lr": 0.0001, "lr_schedule": "cosinedecay", "conv_type": "attention", "ntrain": null, "ntest": 1000, "nvalid": null, "num_workers": 8, "prefetch_factor": 50, "checkpoint_freq": 1, "comet_name": "particleflow-pt", "comet_offline": false, "comet_step_freq": 10, "dtype": "bfloat16", "val_freq": null, "model": {"trainable": "all", "learned_representation_mode": "last", "input_encoding": "split", "pt_mode": "direct-elemtype-split", "eta_mode": "linear", "sin_phi_mode": "linear", "cos_phi_mode": "linear", "energy_mode": "direct-elemtype-split", "gnn_lsh": {"conv_type": "gnn_lsh", "embedding_dim": 512, "width": 512, "num_convs": 8, "dropout_ff": 0.0, "activation": "elu", "bin_size": 320, "max_num_bins": 200, "distance_dim": 128, "layernorm": true, "num_node_messages": 2, "ffn_dist_hidden_dim": 128, "ffn_dist_num_layers": 2}, "attention": {"conv_type": "attention", "num_convs": 3, "dropout_ff": 0.0, "dropout_conv_id_mha": 0.0, "dropout_conv_id_ff": 0.0, "dropout_conv_reg_mha": 0.0, "dropout_conv_reg_ff": 0.0, "activation": "relu", "head_dim": 16, "num_heads": 16, "attention_type": "flash", "use_pre_layernorm": true}, "mamba": {"conv_type": "mamba", "embedding_dim": 1024, "width": 1024, "num_convs": 4, "dropout_ff": 0.0, "activation": "elu", "d_state": 32, "d_conv": 4, "expand": 2}}, "lr_schedule_config": {"onecycle": {"pct_start": 0.3}}, "raytune": {"local_dir": null, "sched": "asha", "search_alg": "hyperopt", "default_metric": "val_loss", "default_mode": "min", "asha": {"max_t": 200, "reduction_factor": 4, "brackets": 1, "grace_period": 10}, "hyperband": {"max_t": 200, "reduction_factor": 4}, "hyperopt": {"n_random_steps": 10}, "nevergrad": {"n_random_steps": 10}}, "train_dataset": {"cms": {"physical_pu": {"batch_size": 1, "samples": {"cms_pf_ttbar": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_qcd": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_ztt": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}}}, "physical_nopu": {"batch_size": 8, "samples": {"cms_pf_ttbar_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_qcd_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_ztt_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}}}}}, "valid_dataset": {"cms": {"physical_pu": {"batch_size": 1, "samples": {"cms_pf_ttbar": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_qcd": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_ztt": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}}}, "physical_nopu": {"batch_size": 8, "samples": {"cms_pf_ttbar_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_qcd_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_ztt_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}}}}}, "test_dataset": {"cms_pf_ttbar": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_qcd": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_ztt": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_ttbar_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_qcd_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_ztt_nopu": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "cms_pf_single_ele": {"version": "2.5.0", "splits": [1]}}, "enabled_test_datasets": ["cms_pf_qcd"], "start_epoch": 1}

cms/v2.4.0/pyg-cms_20250410_144744_012331/model_kwargs.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e67fcae0627f6e53393e8e0fa2949bc8a5b1c1c58dfe79969dffa548562f600
+size 565

cms/v2.4.0/pyg-cms_20250410_144744_012331/overridden_config.yaml

@@ -0,0 +1,353 @@
+checkpoint_freq: 1
+comet: true
+comet_name: particleflow-pt
+comet_offline: false
+comet_step_freq: 10
+conv_type: attention
+data_dir: /scratch/persistent/joosep/tensorflow_datasets
+dataset: cms
+dtype: bfloat16
+enabled_test_datasets:
+- cms_pf_qcd
+finetune: null
+gpu_batch_multiplier: 5
+gpus: 1
+load: null
+lr: 0.0001
+lr_schedule: cosinedecay
+lr_schedule_config:
+  onecycle:
+    pct_start: 0.3
+make_plots: null
+model:
+  attention:
+    activation: relu
+    attention_type: flash
+    conv_type: attention
+    dropout_conv_id_ff: 0.0
+    dropout_conv_id_mha: 0.0
+    dropout_conv_reg_ff: 0.0
+    dropout_conv_reg_mha: 0.0
+    dropout_ff: 0.0
+    head_dim: 16
+    num_convs: 3
+    num_heads: 16
+    use_pre_layernorm: true
+  cos_phi_mode: linear
+  energy_mode: direct-elemtype-split
+  eta_mode: linear
+  gnn_lsh:
+    activation: elu
+    bin_size: 320
+    conv_type: gnn_lsh
+    distance_dim: 128
+    dropout_ff: 0.0
+    embedding_dim: 512
+    ffn_dist_hidden_dim: 128
+    ffn_dist_num_layers: 2
+    layernorm: true
+    max_num_bins: 200
+    num_convs: 8
+    num_node_messages: 2
+    width: 512
+  input_encoding: split
+  learned_representation_mode: last
+  mamba:
+    activation: elu
+    conv_type: mamba
+    d_conv: 4
+    d_state: 32
+    dropout_ff: 0.0
+    embedding_dim: 1024
+    expand: 2
+    num_convs: 4
+    width: 1024
+  pt_mode: direct-elemtype-split
+  sin_phi_mode: linear
+  trainable: all
+ntest: 1000
+ntrain: null
+num_epochs: 10
+num_workers: 8
+nvalid: null
+patience: 20
+prefetch_factor: 50
+raytune:
+  asha:
+    brackets: 1
+    grace_period: 10
+    max_t: 200
+    reduction_factor: 4
+  default_metric: val_loss
+  default_mode: min
+  hyperband:
+    max_t: 200
+    reduction_factor: 4
+  hyperopt:
+    n_random_steps: 10
+  local_dir: null
+  nevergrad:
+    n_random_steps: 10
+  sched: asha
+  search_alg: hyperopt
+save_attention: false
+sort_data: true
+start_epoch: 1
+test: null
+test_dataset:
+  cms_pf_qcd:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+  cms_pf_qcd_nopu:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+  cms_pf_single_ele:
+    splits:
+    - 1
+    version: 2.5.0
+  cms_pf_ttbar:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+  cms_pf_ttbar_nopu:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+  cms_pf_ztt:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+  cms_pf_ztt_nopu:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+train: true
+train_dataset:
+  cms:
+    physical_nopu:
+      batch_size: 8
+      samples:
+        cms_pf_qcd_nopu:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ttbar_nopu:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ztt_nopu:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+    physical_pu:
+      batch_size: 1
+      samples:
+        cms_pf_qcd:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ttbar:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ztt:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+val_freq: null
+valid_dataset:
+  cms:
+    physical_nopu:
+      batch_size: 8
+      samples:
+        cms_pf_qcd_nopu:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ttbar_nopu:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ztt_nopu:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+    physical_pu:
+      batch_size: 1
+      samples:
+        cms_pf_qcd:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ttbar:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        cms_pf_ztt:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_0.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a922633dd93bc4dc77900feda738ab1b4670f4a88ecb523022bf78c84c1aa3fd
+size 2106353

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_100.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:433e953f4097263c5fab4524f04f0171133fe95f0fe89f9130155dc2892ab321
+size 1736194

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1000.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a586d7195ee9c06b8869c09ab9e05ca9171dbd9ae9c657124a21235085893a65
+size 1553303

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1001.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37b9f2026f1b606d8433382a44e00813a6b1ea705ab40153fe241842bdb76ef0
+size 1900873

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1002.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf833821520a88bc3ee168e14c237afc9c86b88eceabda79dcdcddee1e139f7f
+size 1893209

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1003.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0081c0098e48b4eb67412cb453d577151d7ce98a5ad8a93c2f25b0c6201a7a4
+size 1574481

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1004.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32738c0cc05883aaf3e1fe3d1e28ae56dcd4cf0c060e934997edb634461b5a53
+size 1842189

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1005.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ede9bf40e45eb13fe515f16dcc89743330209221cbc4aebbd22274906e7ac39b
+size 1720880

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1006.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be9c96ea4ccd11b5c60e4d8d8b12bf4131af7f6d433d2657442c3abb8d33480e
+size 2036844

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1007.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fdf647a2d9c315e923f6e0e86957cb9db72164d1c671669f7d3608acfbc3dbb6
+size 2067104

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1008.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc1c993172d04b99007b19bc4946859e48449abee2b0a7877061ab7f83da472f
+size 1537090

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1009.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1663d6da6a947efc589f483d468038a5ce860b21a873115e91b28cb999885cd6
+size 1868973

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_101.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4721b07037295d94ee2f33381a1aaa15e99cd1df68d664ed6fb893d5d83b400e
+size 1992442

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1010.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4eb9c061f98b00b1c1acfccbef37cf84e27d4fb419db383d1d6a686821165364
+size 1551400

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1011.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b910a360cd31e8cb6576eb8868659390f7187ded50a508d90b0abcf9d616d6b
+size 1792634

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1012.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c31de00562d7e0a145a56025076417d102e8ec6e37d32bc651d3bedff0328c40
+size 2017466

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1013.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa4d18d6c93c3e21f1670d96be4eea23de9524934567f03ce49da9dd8f3203f3
+size 1808333

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1014.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:03f533c020924bd3d8fd0fa57604ed2c564c90ae0170118fc2e2e4fb68826faa
+size 1617307

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1015.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:373767c923a3078038818debc4208c9f62df86adf26ad6cbd6c3c392910a2a7b
+size 1831373

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1016.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6384e54337ffc10318ea0edb10efeb16d5f67c9529922c382bc6cf7e0d1abfc3
+size 1866554

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1017.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fcfdae71e1fd151db477c59d8991ede90eea80b64598f46b6855aba50d7d0011
+size 1889506

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1018.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c492ce24e9358c3afe548e483bd063d0d6f041e003c5c29f0410bc743e47d08c
+size 2217165

cms/v2.4.0/pyg-cms_20250410_144744_012331/preds_checkpoint-10-3.539432/cms_pf_qcd/pred_0_1019.parquet

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a15ca99909ea3527b5f2fca34e407d195917446c45e51e904ca8162a79ca0027
+size 2172580