added clic v2.2.0 model

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/.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.2.0/pyg-clic_20250106_193536_269746/checkpoints/checkpoint-05-1.995116.pth

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/history/epoch_1.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.1305352703831158, "Regression_eta": 0.0001703609830854827, "Regression_sin_phi": 0.00011389742359422293, "Regression_cos_phi": 0.00011350113057241124, "Regression_energy": 0.13012066721346552, "Classification_binary": 2.3982768406913437, "Classification": 0.019027657360685162, "MET": 5.1761743545396195, "Sliced_Wasserstein_Loss": 44.2138250956898, "Total": 2.6783336899412786}, "valid": {"Regression_pt": 0.11193011826572029, "Regression_eta": 0.00015753862806038385, "Regression_sin_phi": 0.00010198930552825324, "Regression_cos_phi": 0.00010130293065498918, "Regression_energy": 0.11214985005215578, "Classification_binary": 2.1906955858163046, "Classification": 0.016067762814627188, "MET": 3.8647029962822295, "Sliced_Wasserstein_Loss": 40.229381753503795, "Total": 2.431203068163582}, "epoch_train_time": 132967.39482712746, "epoch_valid_time": 4570.915577888489, "epoch_total_time": 137538.31040644646}

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/history/epoch_2.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.10480801537648884, "Regression_eta": 0.00014531593974815654, "Regression_sin_phi": 9.258517870290743e-05, "Regression_cos_phi": 9.23779745192392e-05, "Regression_energy": 0.10448024252683474, "Classification_binary": 2.0554677656134843, "Classification": 0.015208731034772247, "MET": 3.69504114364168, "Sliced_Wasserstein_Loss": 39.60444749542353, "Total": 2.2802871585478925}, "valid": {"Regression_pt": 0.10290886489314753, "Regression_eta": 0.00014659665135173118, "Regression_sin_phi": 9.237539580738888e-05, "Regression_cos_phi": 9.211300018064042e-05, "Regression_energy": 0.10316577804041403, "Classification_binary": 2.0031310179736814, "Classification": 0.014308269513384141, "MET": 3.4699974757943726, "Sliced_Wasserstein_Loss": 38.35105849470418, "Total": 2.2238372773349737}, "epoch_train_time": 133856.32013082504, "epoch_valid_time": 4569.6252617836, "epoch_total_time": 138425.94539380074}

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/history/epoch_3.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.09813356137668497, "Regression_eta": 0.00013694942749972418, "Regression_sin_phi": 8.56519779856205e-05, "Regression_cos_phi": 8.55102180383193e-05, "Regression_energy": 0.0979228353340525, "Classification_binary": 1.9320134381760692, "Classification": 0.013767266464514538, "MET": 3.4966348839835484, "Sliced_Wasserstein_Loss": 38.183917576016924, "Total": 2.142123867769774}, "valid": {"Regression_pt": 0.09853313328098254, "Regression_eta": 0.0001411075526741822, "Regression_sin_phi": 8.771219801441085e-05, "Regression_cos_phi": 8.797908296071872e-05, "Regression_energy": 0.09849909281136328, "Classification_binary": 1.9181835582272386, "Classification": 0.013032377958680341, "MET": 3.3652922562051995, "Sliced_Wasserstein_Loss": 38.05247004386434, "Total": 2.1285604254707393}, "epoch_train_time": 133805.38659882545, "epoch_valid_time": 4571.792712926865, "epoch_total_time": 138377.17933630943}

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/history/epoch_4.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.0939493963110394, "Regression_eta": 0.00013135263595216482, "Regression_sin_phi": 8.168683719655415e-05, "Regression_cos_phi": 8.153408107581176e-05, "Regression_energy": 0.09374886818767385, "Classification_binary": 1.8394567746594395, "Classification": 0.012325539255755504, "MET": 3.375277857023988, "Sliced_Wasserstein_Loss": 37.20878800846349, "Total": 2.039784259622471}, "valid": {"Regression_pt": 0.09506831888949997, "Regression_eta": 0.00013458357656194627, "Regression_sin_phi": 8.367013966984911e-05, "Regression_cos_phi": 8.353278179582944e-05, "Regression_energy": 0.09523754174969241, "Classification_binary": 1.8408782981839094, "Classification": 0.012000219725256516, "MET": 3.4283472471113727, "Sliced_Wasserstein_Loss": 36.69639657109233, "Total": 2.0434852927409866}, "epoch_train_time": 133771.94926667213, "epoch_valid_time": 4636.674633026123, "epoch_total_time": 138408.6239004135}

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/history/epoch_5.json

@@ -0,0 +1 @@
+{"train": {"Regression_pt": 0.09087318845021158, "Regression_eta": 0.00012659103211811747, "Regression_sin_phi": 7.860582842907444e-05, "Regression_cos_phi": 7.848353851052094e-05, "Regression_energy": 0.09071914008637341, "Classification_binary": 1.7849193471697216, "Classification": 0.01164742623439283, "MET": 3.2853502484368686, "Sliced_Wasserstein_Loss": 36.545714166844974, "Total": 1.9784337630459072}, "valid": {"Regression_pt": 0.0924317333498549, "Regression_eta": 0.00013063854873786193, "Regression_sin_phi": 8.12281475126316e-05, "Regression_cos_phi": 8.116436601514114e-05, "Regression_energy": 0.09261981461715657, "Classification_binary": 1.7982992537712634, "Classification": 0.011474268186713484, "MET": 3.238939256913983, "Sliced_Wasserstein_Loss": 36.15146236760458, "Total": 1.9951164561490318}, "epoch_train_time": 133846.90239071846, "epoch_valid_time": 4573.774159908295, "epoch_total_time": 138420.676551342}

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/hyperparameters.json

@@ -0,0 +1 @@
+{"num_mlpf_params": 51577874, "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": "experiments/pyg-clic_20250106_193536_269746/checkpoints/checkpoint-01-2.431203.pth", "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": "float32", "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": "math", "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.2.0/pyg-clic_20250106_193536_269746/model_kwargs.pkl

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/overridden_config.yaml

@@ -0,0 +1,224 @@
+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: float32
+enabled_test_datasets:
+- clic_edm_qq_pf
+gpu_batch_multiplier: 256
+gpus: 1
+load: experiments/pyg-clic_20250106_193536_269746/checkpoints/checkpoint-01-2.431203.pth
+lr: 0.0001
+lr_schedule: cosinedecay
+lr_schedule_config:
+  onecycle:
+    pct_start: 0.3
+make_plots: null
+model:
+  attention:
+    activation: relu
+    attention_type: math
+    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.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_eta.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_pt.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_pt_log.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_res.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_res_bins_0p5_1p5.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/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:aa1796a0199148137522ca007c12a027d2c789a281889f3c69e615573c29d5a1
+size 31596

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_res_vstarget.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_10.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_100.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_20.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_200.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_40.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_400.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_60.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_80.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_binned_eta.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_iqr_over_med_eta.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_iqr_over_med_pt.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_iqr_over_med_pt_vstarget.pdf

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

clic/clusters/v2.2.0/pyg-clic_20250106_193536_269746/plots_checkpoint-05-1.995116/clic_edm_qq_pf/jet_response_med_eta.pdf

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