added v2.3.0 clic model

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/.gitattributes

@@ -0,0 +1,4 @@
+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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/attn_conv_id_0.npz

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/attn_conv_id_1.npz

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/attn_conv_id_2.npz

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/attn_conv_reg_0.npz

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/attn_conv_reg_1.npz

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/attn_conv_reg_2.npz

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch1.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch10.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch2.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch3.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch4.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch5.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch6.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch7.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch8.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/batch0_epoch9.parquet

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/checkpoints/checkpoint-10-1.932789.pth

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_1.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.1303794517487162, "Regression_eta": 0.00016919281536940543, "Regression_sin_phi": 0.00011360028884982376, "Regression_cos_phi": 0.00011387196235138897, "Regression_energy": 0.12985570704583008, "Classification_binary": 2.392060161246226, "Classification": 0.019081550490948843, "ispu": 1.1339613978857038e-05, "MET": 4.746614824786629, "Sliced_Wasserstein_Loss": 44.227388084539854, "Total": 2.671776747081758}, "valid": {"Regression_pt": 0.11279405271661362, "Regression_eta": 0.00015722963023570894, "Regression_sin_phi": 0.00010127424109717977, "Regression_cos_phi": 0.00010164024817875978, "Regression_energy": 0.1131029570818177, "Classification_binary": 2.1817085361613353, "Classification": 0.016212521331940562, "ispu": 1.038243703770257e-13, "MET": 3.884641713250241, "Sliced_Wasserstein_Loss": 39.98020086658821, "Total": 2.4241848153150745}, "epoch_train_time": 26285.219334602356, "epoch_valid_time": 1022.8731498718262, "epoch_total_time": 27308.092485427856}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_10.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.0835242381498437, "Regression_eta": 0.00011547015980284595, "Regression_sin_phi": 7.18869340631474e-05, "Regression_cos_phi": 7.191251433672478e-05, "Regression_energy": 0.08351391834309548, "Classification_binary": 1.7048608634667046, "Classification": 0.010803679734304792, "ispu": 0.0, "MET": 3.0622475884387703, "Sliced_Wasserstein_Loss": 35.204196681168725, "Total": 1.8829506855787748}, "valid": {"Regression_pt": 0.0883274308956918, "Regression_eta": 0.00012227076316907774, "Regression_sin_phi": 7.615695729719402e-05, "Regression_cos_phi": 7.625768002793408e-05, "Regression_energy": 0.08875586543476517, "Classification_binary": 1.7445464203086551, "Classification": 0.01088314805296611, "ispu": 0.0, "MET": 3.014364986760458, "Sliced_Wasserstein_Loss": 34.98203634856104, "Total": 1.9327894728923718}, "epoch_train_time": 26290.289536237717, "epoch_valid_time": 1019.9908812046051, "epoch_total_time": 27310.280418395996}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_2.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.10535963168342724, "Regression_eta": 0.0001447714817914207, "Regression_sin_phi": 9.284823097776953e-05, "Regression_cos_phi": 9.285914779738177e-05, "Regression_energy": 0.10502292041847942, "Classification_binary": 2.060908079844519, "Classification": 0.015259565369672426, "ispu": 3.7658625633433564e-13, "MET": 3.683681694719825, "Sliced_Wasserstein_Loss": 39.72777738154673, "Total": 2.2868949923329294}, "valid": {"Regression_pt": 0.10395971175577726, "Regression_eta": 0.0001448192889261618, "Regression_sin_phi": 9.188002107538003e-05, "Regression_cos_phi": 9.216508011365429e-05, "Regression_energy": 0.10436990170016115, "Classification_binary": 2.017542393281267, "Classification": 0.014487285620135472, "ispu": 6.389157120584089e-14, "MET": 3.5653593733283406, "Sliced_Wasserstein_Loss": 38.48706804322242, "Total": 2.2406844692815877}, "epoch_train_time": 26363.406599521637, "epoch_valid_time": 1023.4262483119965, "epoch_total_time": 27386.832849025726}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_3.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.09844725332016856, "Regression_eta": 0.00013559268107726105, "Regression_sin_phi": 8.50885091187188e-05, "Regression_cos_phi": 8.516677466659304e-05, "Regression_energy": 0.09828658240318094, "Classification_binary": 1.9348388132087584, "Classification": 0.013632855044535132, "ispu": 1.3515285988103442e-13, "MET": 3.481814109193353, "Sliced_Wasserstein_Loss": 38.24706689965053, "Total": 2.1455203727147376}, "valid": {"Regression_pt": 0.09897175449425617, "Regression_eta": 0.0001384762630011674, "Regression_sin_phi": 8.6797830751647e-05, "Regression_cos_phi": 8.699111381392817e-05, "Regression_energy": 0.09918942282617417, "Classification_binary": 1.9264248388520382, "Classification": 0.012931299704440178, "ispu": 6.19452779075414e-14, "MET": 3.45955294813844, "Sliced_Wasserstein_Loss": 37.26410546164545, "Total": 2.1378301945142826}, "epoch_train_time": 26439.678246974945, "epoch_valid_time": 1025.9998333454132, "epoch_total_time": 27465.678081035614}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_4.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.09415529328774695, "Regression_eta": 0.00013005900983250641, "Regression_sin_phi": 8.094461249366533e-05, "Regression_cos_phi": 8.105386571174732e-05, "Regression_energy": 0.09399966619081794, "Classification_binary": 1.860056826961938, "Classification": 0.012479565853972018, "ispu": 2.6953727639999444e-14, "MET": 3.3718585068825333, "Sliced_Wasserstein_Loss": 37.33250421986069, "Total": 2.0609938886313386}, "valid": {"Regression_pt": 0.09546226892443765, "Regression_eta": 0.00013412289840340728, "Regression_sin_phi": 8.333051371755506e-05, "Regression_cos_phi": 8.351954340462865e-05, "Regression_energy": 0.09567689431393428, "Classification_binary": 1.8732029661923613, "Classification": 0.012276103104501277, "ispu": 1.6744395060136565e-14, "MET": 3.3562696837889163, "Sliced_Wasserstein_Loss": 36.757629453300524, "Total": 2.0769232858804965}, "epoch_train_time": 26392.706959486008, "epoch_valid_time": 1019.5876340866089, "epoch_total_time": 27412.294598817825}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_5.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.09128969538623612, "Regression_eta": 0.00012598137932276273, "Regression_sin_phi": 7.820913863811798e-05, "Regression_cos_phi": 7.830209598061055e-05, "Regression_energy": 0.09114466669601252, "Classification_binary": 1.8210710348168224, "Classification": 0.012003366700553708, "ispu": 1.7574601113435228e-14, "MET": 3.2876158232888764, "Sliced_Wasserstein_Loss": 36.762427786891095, "Total": 2.015791788507715}, "valid": {"Regression_pt": 0.09345770774964561, "Regression_eta": 0.00013018650486722557, "Regression_sin_phi": 8.114772874989891e-05, "Regression_cos_phi": 8.149943250225765e-05, "Regression_energy": 0.0938230240881934, "Classification_binary": 1.8401003497111372, "Classification": 0.011863977039027395, "ispu": 1.2271120961840451e-14, "MET": 3.2242925955520487, "Sliced_Wasserstein_Loss": 35.94004092222103, "Total": 2.0395380870867656}, "epoch_train_time": 26359.15105319023, "epoch_valid_time": 1018.8329827785492, "epoch_total_time": 27377.984051704407}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_6.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.08908143086709222, "Regression_eta": 0.00012289961669390793, "Regression_sin_phi": 7.625550405050093e-05, "Regression_cos_phi": 7.636160011981655e-05, "Regression_energy": 0.08895419194496945, "Classification_binary": 1.7890727153317643, "Classification": 0.011638178021750855, "ispu": 4.94479142958256e-15, "MET": 3.222524472220241, "Sliced_Wasserstein_Loss": 36.317898152770844, "Total": 1.9790110221572403}, "valid": {"Regression_pt": 0.09169630771194968, "Regression_eta": 0.0001276085386177276, "Regression_sin_phi": 7.96144666984985e-05, "Regression_cos_phi": 7.973148234326457e-05, "Regression_energy": 0.09199030427125682, "Classification_binary": 1.802469416992083, "Classification": 0.011491252506966641, "ispu": 5.993303133950713e-15, "MET": 3.1848466586872792, "Sliced_Wasserstein_Loss": 35.71983323526265, "Total": 1.9979275603134696}, "epoch_train_time": 26240.556960582733, "epoch_valid_time": 1025.61705493927, "epoch_total_time": 27266.174018859863}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_7.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.08730732955626085, "Regression_eta": 0.00012042917329081596, "Regression_sin_phi": 7.483660657557249e-05, "Regression_cos_phi": 7.49533700667522e-05, "Regression_energy": 0.08720602945098721, "Classification_binary": 1.7599814935335092, "Classification": 0.011337872661106927, "ispu": 4.152324562030984e-16, "MET": 3.160498835080712, "Sliced_Wasserstein_Loss": 35.95462758243587, "Total": 1.9461367844067232}, "valid": {"Regression_pt": 0.08974938613534089, "Regression_eta": 0.0001249850032239518, "Regression_sin_phi": 7.828300721354314e-05, "Regression_cos_phi": 7.840816612688503e-05, "Regression_energy": 0.09003624549278846, "Classification_binary": 1.7810124154140365, "Classification": 0.011250955555749422, "ispu": 0.0, "MET": 3.0956130643521984, "Sliced_Wasserstein_Loss": 35.52940448807104, "Total": 1.9723277689365573}, "epoch_train_time": 26304.619508743286, "epoch_valid_time": 1027.3716974258423, "epoch_total_time": 27331.99122428894}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_8.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.08572647214074722, "Regression_eta": 0.0001182898734159619, "Regression_sin_phi": 7.36364535982949e-05, "Regression_cos_phi": 7.372297761251131e-05, "Regression_energy": 0.08565631138485724, "Classification_binary": 1.7354710122316526, "Classification": 0.011093407002823517, "ispu": 0.0, "MET": 3.1128757013289587, "Sliced_Wasserstein_Loss": 35.63034020397974, "Total": 1.9182216258944915}, "valid": {"Regression_pt": 0.08909278968393937, "Regression_eta": 0.0001240000057006196, "Regression_sin_phi": 7.724219290881383e-05, "Regression_cos_phi": 7.74255981926894e-05, "Regression_energy": 0.08943940433344255, "Classification_binary": 1.760857218559431, "Classification": 0.01103433527588857, "ispu": 0.0, "MET": 3.1605476272467103, "Sliced_Wasserstein_Loss": 35.36565542420028, "Total": 1.9507052730822725}, "epoch_train_time": 26272.195580005646, "epoch_valid_time": 1018.5712554454803, "epoch_total_time": 27290.766838788986}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/history/epoch_9.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.0842812464710672, "Regression_eta": 0.00011656210249807929, "Regression_sin_phi": 7.25083198889265e-05, "Regression_cos_phi": 7.255271835622798e-05, "Regression_energy": 0.08424323498940577, "Classification_binary": 1.7160000326890616, "Classification": 0.010906938745527542, "ispu": 0.0, "MET": 3.078275258243587, "Sliced_Wasserstein_Loss": 35.35614435489623, "Total": 1.8956750885576397}, "valid": {"Regression_pt": 0.08857691952464694, "Regression_eta": 0.0001229213318927308, "Regression_sin_phi": 7.659502305920218e-05, "Regression_cos_phi": 7.65970700358763e-05, "Regression_energy": 0.08898281278209251, "Classification_binary": 1.7537355317882743, "Classification": 0.010974231449387169, "ispu": 0.0, "MET": 3.08492802670643, "Sliced_Wasserstein_Loss": 35.037197763988445, "Total": 1.942550500829143}, "epoch_train_time": 26288.19974923134, "epoch_valid_time": 1024.5263421535492, "epoch_total_time": 27312.726093053818}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/hyperparameters.json

@@ -0,0 +1 @@
+{"num_mlpf_params": 52630547, "train": true, "test": null, "make_plots": null, "comet": true, "save_attention": true, "dataset": "clic", "sort_data": false, "data_dir": "/scratch/persistent/joosep/tensorflow_datasets", "gpus": 1, "gpu_batch_multiplier": 256, "load": null, "finetune": null, "num_epochs": 10, "patience": 20, "lr": 0.0001, "lr_schedule": "cosinedecay", "conv_type": "attention", "ntrain": null, "ntest": 2000, "nvalid": null, "num_workers": 8, "prefetch_factor": 100, "checkpoint_freq": 1, "comet_name": "particleflow-pt", "comet_offline": false, "comet_step_freq": 100, "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, "activation": "elu", "bin_size": 32, "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": 32, "num_heads": 32, "attention_type": "flash", "use_pre_layernorm": true}, "mamba": {"conv_type": "mamba", "embedding_dim": 128, "width": 128, "num_convs": 2, "dropout": 0.0, "activation": "elu", "num_heads": 2, "d_state": 16, "d_conv": 4, "expand": 2}}, "lr_schedule_config": {"onecycle": {"pct_start": 0.3}}, "raytune": {"local_dir": null, "sched": null, "search_alg": null, "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": {"clic": {"physical": {"batch_size": 1, "samples": {"clic_edm_qq_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "clic_edm_ttbar_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "clic_edm_ww_fullhad_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}}}}}, "valid_dataset": {"clic": {"physical": {"batch_size": 1, "samples": {"clic_edm_qq_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "clic_edm_ttbar_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "clic_edm_ww_fullhad_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}}}}}, "test_dataset": {"clic_edm_qq_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "clic_edm_ttbar_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}, "clic_edm_ww_fullhad_pf": {"version": "2.5.0", "splits": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}}, "enabled_test_datasets": ["clic_edm_qq_pf"]}

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/model_kwargs.pkl

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/overridden_config.yaml

@@ -0,0 +1,225 @@
+checkpoint_freq: 1
+comet: true
+comet_name: particleflow-pt
+comet_offline: false
+comet_step_freq: 100
+conv_type: attention
+data_dir: /scratch/persistent/joosep/tensorflow_datasets
+dataset: clic
+dtype: bfloat16
+enabled_test_datasets:
+- clic_edm_qq_pf
+finetune: null
+gpu_batch_multiplier: 256
+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: 32
+    num_convs: 3
+    num_heads: 32
+    use_pre_layernorm: true
+  cos_phi_mode: linear
+  energy_mode: direct-elemtype-split
+  eta_mode: linear
+  gnn_lsh:
+    activation: elu
+    bin_size: 32
+    conv_type: gnn_lsh
+    distance_dim: 128
+    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: 16
+    dropout: 0.0
+    embedding_dim: 128
+    expand: 2
+    num_convs: 2
+    num_heads: 2
+    width: 128
+  pt_mode: direct-elemtype-split
+  sin_phi_mode: linear
+  trainable: all
+ntest: 2000
+ntrain: null
+num_epochs: 10
+num_workers: 8
+nvalid: null
+patience: 20
+prefetch_factor: 100
+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: null
+  search_alg: null
+save_attention: true
+sort_data: false
+test: null
+test_dataset:
+  clic_edm_qq_pf:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+  clic_edm_ttbar_pf:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+  clic_edm_ww_fullhad_pf:
+    splits:
+    - 1
+    - 2
+    - 3
+    - 4
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    version: 2.5.0
+train: true
+train_dataset:
+  clic:
+    physical:
+      batch_size: 1
+      samples:
+        clic_edm_qq_pf:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        clic_edm_ttbar_pf:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        clic_edm_ww_fullhad_pf:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+val_freq: null
+valid_dataset:
+  clic:
+    physical:
+      batch_size: 1
+      samples:
+        clic_edm_qq_pf:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        clic_edm_ttbar_pf:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0
+        clic_edm_ww_fullhad_pf:
+          splits:
+          - 1
+          - 2
+          - 3
+          - 4
+          - 5
+          - 6
+          - 7
+          - 8
+          - 9
+          - 10
+          version: 2.5.0

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_eta.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_pt.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_pt_log.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_res.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_res_bins_0p5_1p5.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_res_bins_0p5_1p5_vstarget.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_res_vstarget.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_response_binned_10.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_response_binned_100.pdf

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

clic/clusters/v2.3.0/pyg-clic_20250130_214007_333962/plots_checkpoint-10-1.932789/clic_edm_qq_pf/jet_response_binned_20.pdf

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