update cms model

README.md

@@ -10,3 +10,4 @@ tags:
 Please see the linked model cards below for more details.
   - [clic/clusters/v1.6](clic/clusters/v1.6/README.md)
   - [cms/2024_04_05](cms/2024_04_05/pyg-cms_20240324_235743_208080/README.md)
+  - [cms/2024_05_16_attn_model21M](cms/2024_05_16_attn_model21M/README.md)

cms/2024_05_16_attn_model21M/README.md

@@ -0,0 +1,94 @@
+# Model Card for mlpf-clic-clusters-v1.6
+
+This model reconstructs particles in a detector, based on the tracks and calorimeter clusters recorded by the detector.
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+- **Developed by:** Joosep Pata, Eric Wulff, Farouk Mokhtar, Mengke Zhang, David Southwick, Maria Girone, David Southwick, Javier Duarte
+- **Model type:** transformer
+- **License:** Apache License
+
+### Model Sources
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** https://github.com/jpata/particleflow/releases/tag/v1.8
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Direct Use
+
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+
+This model may be used to study the physics and computational performance on ML-based reconstruction in simulation in the CMS collaboration.
+
+### Out-of-Scope Use
+
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+
+This model is not intended for physics measurements on real data our use outside the CMS collaboration. 
+
+## Bias, Risks, and Limitations
+
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
+
+The model has only been trained on simulation data and has not been validated against real data.
+It's only meant for internal CMS use.
+
+## Training Details
+
+Approximately 2 weeks on 1x A100 80GB.
+```
+https://www.comet.com/jpata/particleflow-pt/9ceb52e8f9f54d7eb4ef06c9ff85bef2?compareXAxis=step&experiment-tab=panels&showOutliers=true&smoothing=0&xAxis=epoch
+https://www.comet.com/jpata/particleflow-pt/35bb92e72a3846ff98fb563b0769be13?compareXAxis=step&experiment-tab=panels&showOutliers=true&smoothing=0&xAxis=epoch0 
+```
+
+### Training Data
+
+CMS ttbar, QCD, Ztautau with pileup, v1.7.1, 400k events each.
+```
+134G	/eos/user/j/jpata/mlpf/tensorflow_datasets/cms/cms_pf_qcd/1.7.1
+135G	/eos/user/j/jpata/mlpf/tensorflow_datasets/cms/cms_pf_ttbar/1.7.1
+130G	/eos/user/j/jpata/mlpf/tensorflow_datasets/cms/cms_pf_ztt/1.7.1
+```
+### Training Procedure 
+
+```
+#!/bin/bash
+#SBATCH --partition gpu
+#SBATCH --gres gpu:a100:1
+#SBATCH --mem-per-gpu 80G
+
+IMG=/home/software/singularity/pytorch.simg:2024-04-30
+cd ~/particleflow
+git checkout 8bd199fb064bb40558466d906d46498218848e5c
+
+singularity exec --nv \
+    --env PYTHONPATH=hep_tfds \
+    --env KERAS_BACKEND=torch \
+    $IMG python3.10 mlpf/pyg_pipeline.py --dataset cms --gpus 1 \
+    --data-dir /path/to/tensorflow_datasets --config parameters/pytorch/pyg-cms.yaml \
+    --train --conv-type attention --num-epochs 100 --gpu-batch-multiplier 40 --num-workers 4 --prefetch-factor 50 --checkpoint-freq 1 --comet
+```
+
+## Evaluation
+
+```
+WEIGHTS=pyg-cms_20240430_094836_751206/checkpoints/checkpoint-25-17.631161.pth
+singularity exec -B /scratch/persistent --nv \
+     --env PYTHONPATH=hep_tfds \
+     --env KERAS_BACKEND=torch \
+     $IMG  python3.10 mlpf/pyg_pipeline.py --dataset cms --gpus 1 \
+     --data-dir /path/to/joosep/tensorflow_datasets --config parameters/pytorch/pyg-cms.yaml \
+     --test --make-plots --conv-type attention --gpu-batch-multiplier 10 --num-workers 8 --prefetch-factor 10 --load $WEIGHTS --test-datasets cms_pf_ttbar --ntest 50000
+```
+
+## Model Card Contact
+
+Joosep Pata, [email protected]
+

cms/2024_05_16_attn_model21M/{mlpf_cp25_fp32_fused.onnx → onnx/mlpf_21M_attn2x6x512_bs40_relu_tt_qcd_zh400k_checkpoint25_1xa100_fp32_fused.onnx}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f712fe2e3b069c2fd3e390d636e996dd8b493862fe72698b568264178f61fca3
-size 85283193
+oid sha256:370e2cb278d8c7254f925c21b85543baefe6f2c0d9af95bdb51124709a3dbc48
+size 85282119

cms/2024_05_16_attn_model21M/{mlpf_cp25_fp32_unfused.onnx → onnx/mlpf_21M_attn2x6x512_bs40_relu_tt_qcd_zh400k_checkpoint25_1xa100_fp32_unfused.onnx}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b82b9d0d87945324ca95e63cb314261426fef145e289e65dea0b78e5cd585928
-size 85412562
+oid sha256:8e6e1edc7f5830e9fcc91739562b0159f2a4e0f38300cc8b846595e7847822e1
+size 85411488

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_1.json

@@ -0,0 +1 @@
+{"train": {"Classification": 17.48562658807848, "Regression": 3.149008947911026, "MET": 9523.739575957012, "Sliced_Wasserstein_Loss": 2703.3531886533096, "Total": 20.63460787478652, "cls0_Classification": 8.80563741825301, "cls0_Regression": 0.0, "cls1_Classification": 1.6543168073374432, "cls1_Regression": 0.41642663793372764, "cls2_Classification": 1.9072510101220477, "cls2_Regression": 4.416696822801683, "cls3_Classification": 4.140154434123381, "cls3_Regression": 30.493923647269547, "cls4_Classification": 3.753003355792061, "cls4_Regression": 20.56570219102762, "cls5_Classification": 1.0756870717394926, "cls5_Regression": 0.5822255699509518, "cls6_Classification": 0.3781087531110718, "cls6_Regression": 0.04469472770962636}, "valid": {"Classification": 16.269579411029657, "Regression": 3.0924154839011995, "MET": 5732.423192269243, "Sliced_Wasserstein_Loss": 2631.9453515494833, "Total": 19.361958774158612, "cls0_Classification": 8.303593463637121, "cls0_Regression": 0.0, "cls1_Classification": 1.3890330827224258, "cls1_Regression": 0.345031799296068, "cls2_Classification": 1.7639946840636453, "cls2_Regression": 4.34984204483922, "cls3_Classification": 3.9660399658447183, "cls3_Regression": 30.071739107797402, "cls4_Classification": 3.529999114878374, "cls4_Regression": 20.250622188437188, "cls5_Classification": 0.947526117596218, "cls5_Regression": 0.5533835319867336, "cls6_Classification": 0.2997203446832827, "cls6_Regression": 0.0420171756102458}, "epoch_time": 53397.86642074585}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_10.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.892871089682176, "Regression": 3.04007847971425, "MET": 4118.37314116716, "Sliced_Wasserstein_Loss": 2431.152580497355, "Total": 17.93300782063565, "cls0_Classification": 7.458087094389136, "cls0_Regression": 0.0, "cls1_Classification": 1.384684512590078, "cls1_Regression": 0.28774243010502143, "cls2_Classification": 1.551655832153122, "cls2_Regression": 4.161068906985463, "cls3_Classification": 3.796251158516266, "cls3_Regression": 29.84330611904861, "cls4_Classification": 3.375984087974341, "cls4_Regression": 20.09319547423668, "cls5_Classification": 0.7923157828966552, "cls5_Regression": 0.49596719999635525, "cls6_Classification": 0.1718370065241388, "cls6_Regression": 0.03872826724366851}, "valid": {"Classification": 14.884505945934688, "Regression": 3.038905651241253, "MET": 4120.822392535822, "Sliced_Wasserstein_Loss": 2452.5433188937022, "Total": 17.92337970676441, "cls0_Classification": 7.345777969843385, "cls0_Regression": 0.0, "cls1_Classification": 1.3800778972113463, "cls1_Regression": 0.2887544158457597, "cls2_Classification": 1.5650468398554649, "cls2_Regression": 4.158672044214429, "cls3_Classification": 3.8662965209513493, "cls3_Regression": 29.847561229590138, "cls4_Classification": 3.4121253202057646, "cls4_Regression": 20.08999604298567, "cls5_Classification": 0.7796669502093052, "cls5_Regression": 0.4957525551378707, "cls6_Classification": 0.16833817438458015, "cls6_Regression": 0.039230227192335626}, "epoch_time": 53482.81980347633}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_11.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.860461271712417, "Regression": 3.0378164441621194, "MET": 4080.1202982463446, "Sliced_Wasserstein_Loss": 2418.6338984462864, "Total": 17.89822629857958, "cls0_Classification": 7.441826212146458, "cls0_Regression": 0.0, "cls1_Classification": 1.3778686294622402, "cls1_Regression": 0.2826533382516037, "cls2_Classification": 1.5484523394218352, "cls2_Regression": 4.156316386887158, "cls3_Classification": 3.795083855750406, "cls3_Regression": 29.82376494355813, "cls4_Classification": 3.3745196713458574, "cls4_Regression": 20.087512236014497, "cls5_Classification": 0.7905571812388054, "cls5_Regression": 0.49421693737763983, "cls6_Classification": 0.16486340369150143, "cls6_Regression": 0.038514693773398126}, "valid": {"Classification": 14.859540309480174, "Regression": 3.037682426170443, "MET": 4234.071642785738, "Sliced_Wasserstein_Loss": 2403.893368877041, "Total": 17.897200933022326, "cls0_Classification": 7.64115542215095, "cls0_Regression": 0.0, "cls1_Classification": 1.3692719288882873, "cls1_Regression": 0.28695614645378625, "cls2_Classification": 1.5257380026970178, "cls2_Regression": 4.159916095676441, "cls3_Classification": 3.7562781807105963, "cls3_Regression": 29.821473571309564, "cls4_Classification": 3.2919059532239254, "cls4_Regression": 20.092916944351884, "cls5_Classification": 0.7554579788507997, "cls5_Regression": 0.4957560939973134, "cls6_Classification": 0.15529454862702016, "cls6_Regression": 0.039073647915701275}, "epoch_time": 53510.713752031326}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_12.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.831527002540925, "Regression": 3.035432699316866, "MET": 4055.1959011954846, "Sliced_Wasserstein_Loss": 2410.7825217644854, "Total": 17.86699061731995, "cls0_Classification": 7.427402033781814, "cls0_Regression": 0.0, "cls1_Classification": 1.3715849804223768, "cls1_Regression": 0.28088765858004955, "cls2_Classification": 1.545476156693881, "cls2_Regression": 4.151713888136793, "cls3_Classification": 3.7946344425584204, "cls3_Regression": 29.797842816678468, "cls4_Classification": 3.373950826842171, "cls4_Regression": 20.08024222101887, "cls5_Classification": 0.7888911632544674, "cls5_Regression": 0.49251439033406924, "cls6_Classification": 0.15792756043484713, "cls6_Regression": 0.03836854448539749}, "valid": {"Classification": 14.827245084971675, "Regression": 3.0339522242585804, "MET": 4025.9450183272243, "Sliced_Wasserstein_Loss": 2410.427857380873, "Total": 17.86121683397201, "cls0_Classification": 7.443319023867044, "cls0_Regression": 0.0, "cls1_Classification": 1.328606366237296, "cls1_Regression": 0.27249742436115043, "cls2_Classification": 1.5355303115107464, "cls2_Regression": 4.156640494835055, "cls3_Classification": 3.7862457305898034, "cls3_Regression": 29.79502665778074, "cls4_Classification": 3.3918703531114627, "cls4_Regression": 20.08310380914695, "cls5_Classification": 0.8036455296151283, "cls5_Regression": 0.4916388954332931, "cls6_Classification": 0.16321777018337638, "cls6_Regression": 0.03811728640501994}, "epoch_time": 53455.28546309471}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_13.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.804363050360312, "Regression": 3.032954255529637, "MET": 4027.756571000125, "Sliced_Wasserstein_Loss": 2399.227725246803, "Total": 17.83732088557504, "cls0_Classification": 7.41394046528096, "cls0_Regression": 0.0, "cls1_Classification": 1.3656773610301163, "cls1_Regression": 0.2793153632599346, "cls2_Classification": 1.5427433925521723, "cls2_Regression": 4.14758553494814, "cls3_Classification": 3.7942175719581788, "cls3_Regression": 29.764597304952723, "cls4_Classification": 3.3734506472070644, "cls4_Regression": 20.071156329403923, "cls5_Classification": 0.7873750039051111, "cls5_Regression": 0.4910310990023743, "cls6_Classification": 0.15140062501301704, "cls6_Regression": 0.03784379265113431}, "valid": {"Classification": 14.806587908197267, "Regression": 3.0314559860254917, "MET": 4267.6391202932355, "Sliced_Wasserstein_Loss": 2355.942019326891, "Total": 17.838022742419195, "cls0_Classification": 7.449145206806064, "cls0_Regression": 0.0, "cls1_Classification": 1.2995666483567978, "cls1_Regression": 0.2722413248318269, "cls2_Classification": 1.5412460169526825, "cls2_Regression": 4.1465784191727755, "cls3_Classification": 3.810256607172609, "cls3_Regression": 29.75505300316561, "cls4_Classification": 3.412638820913862, "cls4_Regression": 20.075029677607464, "cls5_Classification": 0.7681842868783322, "cls5_Regression": 0.49479995112306313, "cls6_Classification": 0.15109213270055816, "cls6_Regression": 0.037641439466466906}, "epoch_time": 50089.38499498367}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_14.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.77955127671096, "Regression": 3.029446164139626, "MET": 4048.819094430791, "Sliced_Wasserstein_Loss": 2383.1850710209524, "Total": 17.80898800349898, "cls0_Classification": 7.40169143374849, "cls0_Regression": 0.0, "cls1_Classification": 1.3602311142323071, "cls1_Regression": 0.27610711117460535, "cls2_Classification": 1.5402696674407466, "cls2_Regression": 4.143847042008581, "cls3_Classification": 3.7940031162785854, "cls3_Regression": 29.721698358812013, "cls4_Classification": 3.373257669638022, "cls4_Regression": 20.059562054817345, "cls5_Classification": 0.7859203500541508, "cls5_Regression": 0.4897008392083559, "cls6_Classification": 0.14528125504410172, "cls6_Regression": 0.03755252374307494}, "valid": {"Classification": 14.779845520243253, "Regression": 3.0291956717552484, "MET": 4185.503832055982, "Sliced_Wasserstein_Loss": 2387.940686437854, "Total": 17.809074058647116, "cls0_Classification": 7.413807376707764, "cls0_Regression": 0.0, "cls1_Classification": 1.3420541134725925, "cls1_Regression": 0.2757006412901429, "cls2_Classification": 1.5800180864191102, "cls2_Regression": 4.1511631539486835, "cls3_Classification": 3.7779529219426857, "cls3_Regression": 29.72045516077974, "cls4_Classification": 3.388256323413029, "cls4_Regression": 20.07323340136621, "cls5_Classification": 0.7513343533926191, "cls5_Regression": 0.49148839221561563, "cls6_Classification": 0.14505181825784635, "cls6_Regression": 0.03806988806058788}, "epoch_time": 50153.535920619965}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_15.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.756996657224976, "Regression": 3.025392008060149, "MET": 4017.4858999458493, "Sliced_Wasserstein_Loss": 2368.200941392094, "Total": 17.782417628191777, "cls0_Classification": 7.390554057149998, "cls0_Regression": 0.0, "cls1_Classification": 1.3553568848408797, "cls1_Regression": 0.27528351919492233, "cls2_Classification": 1.5382783862519265, "cls2_Regression": 4.140289160453201, "cls3_Classification": 3.7934729974590744, "cls3_Regression": 29.671439580122463, "cls4_Classification": 3.372567766693048, "cls4_Regression": 20.045504956887573, "cls5_Classification": 0.7848426207720665, "cls5_Regression": 0.4885596111941309, "cls6_Classification": 0.13978655029327383, "cls6_Regression": 0.03739054045027883}, "valid": {"Classification": 14.761458420109964, "Regression": 3.0262350700808063, "MET": 4120.437520826391, "Sliced_Wasserstein_Loss": 2401.139453515495, "Total": 17.78771268952016, "cls0_Classification": 7.39570078203099, "cls0_Regression": 0.0, "cls1_Classification": 1.3744199524637621, "cls1_Regression": 0.26884611460893243, "cls2_Classification": 1.5262399512662446, "cls2_Regression": 4.147951659342719, "cls3_Classification": 3.78995087574975, "cls3_Regression": 29.68053357214262, "cls4_Classification": 3.3684998880581474, "cls4_Regression": 20.040257414195267, "cls5_Classification": 0.7801337542433772, "cls5_Regression": 0.48902676972259246, "cls6_Classification": 0.148818984980267, "cls6_Regression": 0.03760942448301738}, "epoch_time": 50099.06511473656}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_16.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.735845274294997, "Regression": 3.0218510486524766, "MET": 4009.937768150956, "Sliced_Wasserstein_Loss": 2349.7359936685134, "Total": 17.757729516391052, "cls0_Classification": 7.379987477610697, "cls0_Regression": 0.0, "cls1_Classification": 1.350771959090057, "cls1_Regression": 0.27236680805338026, "cls2_Classification": 1.5362360131940684, "cls2_Regression": 4.136247370558587, "cls3_Classification": 3.7931062424501185, "cls3_Regression": 29.613144082975797, "cls4_Classification": 3.372103709334777, "cls4_Regression": 20.029182894572415, "cls5_Classification": 0.7836622196130295, "cls5_Regression": 0.48762051335287626, "cls6_Classification": 0.13484802894403194, "cls6_Regression": 0.037322323042335306}, "valid": {"Classification": 14.741458576307897, "Regression": 3.021098761350383, "MET": 4076.047317560813, "Sliced_Wasserstein_Loss": 2347.9886704431856, "Total": 17.762579140286572, "cls0_Classification": 7.347883387829057, "cls0_Regression": 0.0, "cls1_Classification": 1.3738098042839886, "cls1_Regression": 0.26895832085919275, "cls2_Classification": 1.5193760283030657, "cls2_Regression": 4.143401092864878, "cls3_Classification": 3.7940064900241586, "cls3_Regression": 29.619842864878375, "cls4_Classification": 3.3961251197517495, "cls4_Regression": 20.03793787487504, "cls5_Classification": 0.7895646926284988, "cls5_Regression": 0.4894655882934855, "cls6_Classification": 0.1353878744520056, "cls6_Regression": 0.03714336788046563}, "epoch_time": 50199.061577796936}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_17.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.715267942683385, "Regression": 3.0171551526638063, "MET": 4009.843462323489, "Sliced_Wasserstein_Loss": 2333.021202149373, "Total": 17.732419190236183, "cls0_Classification": 7.369300490481943, "cls0_Regression": 0.0, "cls1_Classification": 1.3462052571906111, "cls1_Regression": 0.2706103420031137, "cls2_Classification": 1.5341983588120132, "cls2_Regression": 4.132328917086683, "cls3_Classification": 3.792844925438414, "cls3_Regression": 29.549165347606948, "cls4_Classification": 3.3716969269379766, "cls4_Regression": 20.01092389719665, "cls5_Classification": 0.7824331672793352, "cls5_Regression": 0.4865213059602408, "cls6_Classification": 0.13024812343144707, "cls6_Regression": 0.03733597821891792}, "valid": {"Classification": 14.720500718510497, "Regression": 3.0190223050649783, "MET": 4102.013662112629, "Sliced_Wasserstein_Loss": 2309.365544818394, "Total": 17.739521721926025, "cls0_Classification": 7.320328119793402, "cls0_Regression": 0.0, "cls1_Classification": 1.3769728124479341, "cls1_Regression": 0.26595667655052485, "cls2_Classification": 1.5157938890161613, "cls2_Regression": 4.147709552544985, "cls3_Classification": 3.8043751041319562, "cls3_Regression": 29.55708513828724, "cls4_Classification": 3.3732369158197266, "cls4_Regression": 20.027296890619795, "cls5_Classification": 0.792654564363962, "cls5_Regression": 0.48809255150626873, "cls6_Classification": 0.14984518303469155, "cls6_Regression": 0.03710505565934402}, "epoch_time": 50164.12058401108}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_18.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.694737472403883, "Regression": 3.013311221726996, "MET": 4000.489857125005, "Sliced_Wasserstein_Loss": 2314.7350356146126, "Total": 17.707971893614364, "cls0_Classification": 7.357604032157288, "cls0_Regression": 0.0, "cls1_Classification": 1.3417623798527514, "cls1_Regression": 0.26923309884226476, "cls2_Classification": 1.5322332744095473, "cls2_Regression": 4.128274435581289, "cls3_Classification": 3.792441722726705, "cls3_Regression": 29.48012298496272, "cls4_Classification": 3.37122473393177, "cls4_Regression": 19.990576967134587, "cls5_Classification": 0.7812869358416296, "cls5_Regression": 0.4856349271176115, "cls6_Classification": 0.12623671001231412, "cls6_Regression": 0.03732718663436706}, "valid": {"Classification": 14.701788206014662, "Regression": 3.014447658072309, "MET": 4269.373542152616, "Sliced_Wasserstein_Loss": 2306.0011662779075, "Total": 17.716232609963345, "cls0_Classification": 7.414679481839387, "cls0_Regression": 0.0, "cls1_Classification": 1.2994247685667277, "cls1_Regression": 0.2659569409480694, "cls2_Classification": 1.536039744564312, "cls2_Regression": 4.141946499604298, "cls3_Classification": 3.7920003228090637, "cls3_Regression": 29.51144930856381, "cls4_Classification": 3.3445033296192936, "cls4_Regression": 20.00042303607131, "cls5_Classification": 0.7871707965834305, "cls5_Regression": 0.48859226286550317, "cls6_Classification": 0.13501054864849737, "cls6_Regression": 0.03730152591551514}, "epoch_time": 50153.697790145874}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_19.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.67470295122256, "Regression": 3.007714221372933, "MET": 3970.6965468405047, "Sliced_Wasserstein_Loss": 2297.1456658474613, "Total": 17.682524888574164, "cls0_Classification": 7.345830773316116, "cls0_Regression": 0.0, "cls1_Classification": 1.3375331446296914, "cls1_Regression": 0.2688025806926105, "cls2_Classification": 1.5303394582830008, "cls2_Regression": 4.124196848835756, "cls3_Classification": 3.7920378691631607, "cls3_Regression": 29.40834183779731, "cls4_Classification": 3.370780527554463, "cls4_Regression": 19.969037676511018, "cls5_Classification": 0.7801699113800141, "cls5_Regression": 0.48478121290144544, "cls6_Classification": 0.12255588864706127, "cls6_Regression": 0.03692233645395093}, "valid": {"Classification": 14.681344499750084, "Regression": 3.012300587304232, "MET": 4024.6267910696433, "Sliced_Wasserstein_Loss": 2193.0081639453515, "Total": 17.69367502499167, "cls0_Classification": 7.242916423692103, "cls0_Regression": 0.0, "cls1_Classification": 1.3816999151324558, "cls1_Regression": 0.2658147764221822, "cls2_Classification": 1.5470445398408863, "cls2_Regression": 4.135434347405032, "cls3_Classification": 3.838265825454015, "cls3_Regression": 29.452440332389205, "cls4_Classification": 3.3761298317227593, "cls4_Regression": 20.00554632830723, "cls5_Classification": 0.7672589768254332, "cls5_Regression": 0.48968548569747583, "cls6_Classification": 0.1316297785991909, "cls6_Regression": 0.03729452446515224}, "epoch_time": 50132.245872974396}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_2.json

@@ -0,0 +1 @@
+{"train": {"Classification": 15.809357687341192, "Regression": 3.0819018931978173, "MET": 5034.315657933103, "Sliced_Wasserstein_Loss": 2530.571916524347, "Total": 18.8912686924647, "cls0_Classification": 7.952300370725205, "cls0_Regression": 0.0, "cls1_Classification": 1.553409715135169, "cls1_Regression": 0.3599535259247303, "cls2_Classification": 1.6749120373745157, "cls2_Regression": 4.305406365851627, "cls3_Classification": 3.842675769046528, "cls3_Regression": 29.98076342316824, "cls4_Classification": 3.4391450280126628, "cls4_Regression": 20.202373266130714, "cls5_Classification": 0.8959836258695381, "cls5_Regression": 0.54743448850856, "cls6_Classification": 0.28281637663624154, "cls6_Regression": 0.04322513985992367}, "valid": {"Classification": 15.503986952265912, "Regression": 3.0699831826891035, "MET": 4586.723092302565, "Sliced_Wasserstein_Loss": 2517.696934355215, "Total": 18.574007882789072, "cls0_Classification": 7.642892473342219, "cls0_Regression": 0.0, "cls1_Classification": 1.5750202406489504, "cls1_Regression": 0.3196221173345864, "cls2_Classification": 1.6617523976382873, "cls2_Regression": 4.2866124703223925, "cls3_Classification": 3.8567925274908363, "cls3_Regression": 29.96960388203932, "cls4_Classification": 3.416327370043319, "cls4_Regression": 20.14479808813729, "cls5_Classification": 0.8543495755321143, "cls5_Regression": 0.5393517509137579, "cls6_Classification": 0.25703236453734174, "cls6_Regression": 0.04150927301169793}, "epoch_time": 53426.56957054138}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_20.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.661705440079977, "Regression": 3.003735238680385, "MET": 3998.811013454409, "Sliced_Wasserstein_Loss": 2278.6703878035573, "Total": 17.665403580205773, "cls0_Classification": 7.338745600241596, "cls0_Regression": 0.0, "cls1_Classification": 1.3352637056379388, "cls1_Regression": 0.2676105252179052, "cls2_Classification": 1.5287826206939643, "cls2_Regression": 4.120110150164535, "cls3_Classification": 3.7916421512475527, "cls3_Regression": 29.335342295997002, "cls4_Classification": 3.3703434805473402, "cls4_Regression": 19.946082132294748, "cls5_Classification": 0.7793256751936936, "cls5_Regression": 0.4841094524513683, "cls6_Classification": 0.11990494878446911, "cls6_Regression": 0.0367866135072583}, "valid": {"Classification": 14.679281385371542, "Regression": 3.008955999041986, "MET": 4186.546817727424, "Sliced_Wasserstein_Loss": 2268.216261246251, "Total": 17.688197684105297, "cls0_Classification": 7.328072934021993, "cls0_Regression": 0.0, "cls1_Classification": 1.3613833637485422, "cls1_Regression": 0.26689223709284404, "cls2_Classification": 1.5531831186479508, "cls2_Regression": 4.1347818956181275, "cls3_Classification": 3.7706744236296235, "cls3_Regression": 29.376390161612797, "cls4_Classification": 3.3747315348008997, "cls4_Regression": 19.975259548900368, "cls5_Classification": 0.7726646456389537, "cls5_Regression": 0.4942583835986963, "cls6_Classification": 0.12574103513347112, "cls6_Regression": 0.03704041300198111}, "epoch_time": 50144.622129917145}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_21.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.661430780605656, "Regression": 2.9985170340942227, "MET": 4003.7795642937476, "Sliced_Wasserstein_Loss": 2263.8182196859248, "Total": 17.659915597534052, "cls0_Classification": 7.338925886199858, "cls0_Regression": 0.0, "cls1_Classification": 1.336363726595368, "cls1_Regression": 0.2682260276625047, "cls2_Classification": 1.528365587380764, "cls2_Regression": 4.116453989982088, "cls3_Classification": 3.7914882247886035, "cls3_Regression": 29.261020223268215, "cls4_Classification": 3.3701843472737116, "cls4_Regression": 19.922581174240847, "cls5_Classification": 0.7793294175917441, "cls5_Regression": 0.4837392397920398, "cls6_Classification": 0.11932410420007707, "cls6_Regression": 0.036663561832226015}, "valid": {"Classification": 14.68757940061646, "Regression": 3.004176993085638, "MET": 4096.083638787071, "Sliced_Wasserstein_Loss": 2290.309230256581, "Total": 17.691731662362546, "cls0_Classification": 7.370694143618794, "cls0_Regression": 0.0, "cls1_Classification": 1.308570483016078, "cls1_Regression": 0.26480618114274407, "cls2_Classification": 1.5334494621584471, "cls2_Regression": 4.133956975279074, "cls3_Classification": 3.8017353590469845, "cls3_Regression": 29.323311500333222, "cls4_Classification": 3.390354256914362, "cls4_Regression": 19.940549764661778, "cls5_Classification": 0.7632759295078724, "cls5_Regression": 0.4924220409603049, "cls6_Classification": 0.12445012429125188, "cls6_Regression": 0.03711149780720879}, "epoch_time": 51102.256094932556}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_22.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.651541737826467, "Regression": 2.9940743194693216, "MET": 4007.088932394718, "Sliced_Wasserstein_Loss": 2243.577956429375, "Total": 17.645629399758405, "cls0_Classification": 7.333301224642812, "cls0_Regression": 0.0, "cls1_Classification": 1.3343362417732327, "cls1_Regression": 0.2655604639727475, "cls2_Classification": 1.5271626504769442, "cls2_Regression": 4.112079289373932, "cls3_Classification": 3.791316725330112, "cls3_Regression": 29.18793216561836, "cls4_Classification": 3.3700408344441204, "cls4_Regression": 19.89832652976215, "cls5_Classification": 0.7787186745011871, "cls5_Regression": 0.48309221171168826, "cls6_Classification": 0.11749363426227247, "cls6_Regression": 0.03663010143769916}, "valid": {"Classification": 14.6633465928024, "Regression": 3.0014187979007, "MET": 4002.6321226257915, "Sliced_Wasserstein_Loss": 2217.9905031656112, "Total": 17.66470811812729, "cls0_Classification": 7.260530994876707, "cls0_Regression": 0.0, "cls1_Classification": 1.3934977403365545, "cls1_Regression": 0.26363860158592967, "cls2_Classification": 1.4967720720697268, "cls2_Regression": 4.133581774825059, "cls3_Classification": 3.8408434167777408, "cls3_Regression": 29.26171484505165, "cls4_Classification": 3.3905442977340887, "cls4_Regression": 19.93486025491503, "cls5_Classification": 0.7628149828963262, "cls5_Regression": 0.49135668156187523, "cls6_Classification": 0.11945339712608818, "cls6_Regression": 0.03689439691570909}, "epoch_time": 52108.2375125885}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_23.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.638644457866455, "Regression": 2.9894151966093223, "MET": 3998.2648394218354, "Sliced_Wasserstein_Loss": 2226.471945682509, "Total": 17.62802125421752, "cls0_Classification": 7.325775424876078, "cls0_Regression": 0.0, "cls1_Classification": 1.33112843714854, "cls1_Regression": 0.2648518490375203, "cls2_Classification": 1.5260422090431958, "cls2_Regression": 4.108037174053401, "cls3_Classification": 3.790900831007623, "cls3_Regression": 29.116117174157537, "cls4_Classification": 3.3694700373849296, "cls4_Regression": 19.87427755446328, "cls5_Classification": 0.7782093829403924, "cls5_Regression": 0.48265654776471445, "cls6_Classification": 0.11534358590398112, "cls6_Regression": 0.03654707206573213}, "valid": {"Classification": 14.660187489586804, "Regression": 2.9983055778282237, "MET": 4285.737087637454, "Sliced_Wasserstein_Loss": 2227.126624458514, "Total": 17.658556522825723, "cls0_Classification": 7.221778938270576, "cls0_Regression": 0.0, "cls1_Classification": 1.3330441773315145, "cls1_Regression": 0.2649225770771722, "cls2_Classification": 1.5388303182793235, "cls2_Regression": 4.133360494418527, "cls3_Classification": 3.835530734546818, "cls3_Regression": 29.20544714261913, "cls4_Classification": 3.4273477851132954, "cls4_Regression": 19.903792485838053, "cls5_Classification": 0.7772702881591553, "cls5_Regression": 0.49118531127644954, "cls6_Classification": 0.12874822162143867, "cls6_Regression": 0.03706584346965725}, "epoch_time": 51698.22525763512}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_24.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.630352605490065, "Regression": 2.985456064897738, "MET": 4017.602865830799, "Sliced_Wasserstein_Loss": 2213.3149498063067, "Total": 17.61582559253551, "cls0_Classification": 7.3205660067896865, "cls0_Regression": 0.0, "cls1_Classification": 1.3291068099929186, "cls1_Regression": 0.26498165330033946, "cls2_Classification": 1.5251656743345692, "cls2_Regression": 4.1034912994126715, "cls3_Classification": 3.7908435560461533, "cls3_Regression": 29.045314908151788, "cls4_Classification": 3.3694768713291956, "cls4_Regression": 19.850214260424043, "cls5_Classification": 0.7779767034094223, "cls5_Regression": 0.4821193916097388, "cls6_Classification": 0.11378378897300788, "cls6_Regression": 0.03635459788990581}, "valid": {"Classification": 14.650810406947684, "Regression": 2.99518617492086, "MET": 4218.620126624459, "Sliced_Wasserstein_Loss": 2251.9298567144283, "Total": 17.645977382539154, "cls0_Classification": 7.5285119803815395, "cls0_Regression": 0.0, "cls1_Classification": 1.295908525283239, "cls1_Regression": 0.2653094923762392, "cls2_Classification": 1.490688650658114, "cls2_Regression": 4.137117054731756, "cls3_Classification": 3.7374944029073642, "cls3_Regression": 29.16967781572809, "cls4_Classification": 3.3039628066686104, "cls4_Regression": 19.877650158280574, "cls5_Classification": 0.7736203003946601, "cls5_Regression": 0.49378800002863626, "cls6_Classification": 0.12028372744566915, "cls6_Regression": 0.036913410658361276}, "epoch_time": 51023.51962018013}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_25.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.619906433540217, "Regression": 2.9805860660432373, "MET": 4019.101428749948, "Sliced_Wasserstein_Loss": 2196.3638938642894, "Total": 17.60049412671304, "cls0_Classification": 7.31461331590786, "cls0_Regression": 0.0, "cls1_Classification": 1.327184437742117, "cls1_Regression": 0.26366528512517184, "cls2_Classification": 1.5237591835193902, "cls2_Regression": 4.099088612175615, "cls3_Classification": 3.790360298558754, "cls3_Regression": 28.976142375140583, "cls4_Classification": 3.368794127754405, "cls4_Regression": 19.826463375265547, "cls5_Classification": 0.7773871130034989, "cls5_Regression": 0.48154082500676887, "cls6_Classification": 0.11223242260720831, "cls6_Regression": 0.03640756604274925}, "valid": {"Classification": 14.638349716761079, "Regression": 2.9928012277157614, "MET": 4421.007664111963, "Sliced_Wasserstein_Loss": 2241.7280906364545, "Total": 17.631160706847716, "cls0_Classification": 7.403679893160613, "cls0_Regression": 0.0, "cls1_Classification": 1.3256552828744586, "cls1_Regression": 0.2624994142577474, "cls2_Classification": 1.5186383184771743, "cls2_Regression": 4.133219916277907, "cls3_Classification": 3.7855321012787404, "cls3_Regression": 29.12222748667111, "cls4_Classification": 3.3088609135496503, "cls4_Regression": 19.859552024325225, "cls5_Classification": 0.775150063910988, "cls5_Regression": 0.490803439869002, "cls6_Classification": 0.11909470118033572, "cls6_Regression": 0.036988791923688195}, "epoch_time": 56564.72025060654}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_3.json

@@ -0,0 +1 @@
+{"train": {"Classification": 15.385019213146165, "Regression": 3.068172824592827, "MET": 4669.995917857292, "Sliced_Wasserstein_Loss": 2499.799058607906, "Total": 18.453290967217896, "cls0_Classification": 7.741056644936894, "cls0_Regression": 0.0, "cls1_Classification": 1.5016315228370891, "cls1_Regression": 0.3368930318500854, "cls2_Classification": 1.6061807819073604, "cls2_Regression": 4.264839747261216, "cls3_Classification": 3.806384270733536, "cls3_Regression": 29.936997542383473, "cls4_Classification": 3.388371425521723, "cls4_Regression": 20.14372110217853, "cls5_Classification": 0.8316155183904694, "cls5_Regression": 0.5373879802635689, "cls6_Classification": 0.24416903875041654, "cls6_Regression": 0.042679959922171135}, "valid": {"Classification": 15.299683959513496, "Regression": 3.0630935521492835, "MET": 5496.996334555149, "Sliced_Wasserstein_Loss": 2509.2277574141954, "Total": 18.362813957847383, "cls0_Classification": 7.507295745168277, "cls0_Regression": 0.0, "cls1_Classification": 1.4943239948662945, "cls1_Regression": 0.31332595759095927, "cls2_Classification": 1.6102471051316227, "cls2_Regression": 4.245313085742253, "cls3_Classification": 3.865831181272909, "cls3_Regression": 29.967500416527823, "cls4_Classification": 3.48607885913029, "cls4_Regression": 20.13296349133622, "cls5_Classification": 0.8166237068112713, "cls5_Regression": 0.5346810259210472, "cls6_Classification": 0.23275981573850382, "cls6_Regression": 0.040671420073517164}, "epoch_time": 53540.957287073135}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_4.json

@@ -0,0 +1 @@
+{"train": {"Classification": 15.230913119090266, "Regression": 3.060871894135044, "MET": 4498.631899029449, "Sliced_Wasserstein_Loss": 2494.308160119965, "Total": 18.291770369059023, "cls0_Classification": 7.6534552422210185, "cls0_Regression": 0.0, "cls1_Classification": 1.4680591207772733, "cls1_Regression": 0.3301080438452847, "cls2_Classification": 1.5863162957679011, "cls2_Regression": 4.232781063856375, "cls3_Classification": 3.8015419330820177, "cls3_Regression": 29.916821135502147, "cls4_Classification": 3.3825228709334776, "cls4_Regression": 20.127453711417502, "cls5_Classification": 0.8158737718425876, "cls5_Regression": 0.5275295681988587, "cls6_Classification": 0.2245040386983484, "cls6_Regression": 0.04099524774431025}, "valid": {"Classification": 15.17060068518827, "Regression": 3.0563695565019993, "MET": 4359.173942019327, "Sliced_Wasserstein_Loss": 2555.19476841053, "Total": 18.22696991627791, "cls0_Classification": 7.5751891296651115, "cls0_Regression": 0.0, "cls1_Classification": 1.4658456946642786, "cls1_Regression": 0.32102147184662405, "cls2_Classification": 1.5694979928565478, "cls2_Regression": 4.218745769639287, "cls3_Classification": 3.8752697668485503, "cls3_Regression": 29.924275241586137, "cls4_Classification": 3.333808326911863, "cls4_Regression": 20.117298140203264, "cls5_Classification": 0.8176529861140036, "cls5_Regression": 0.5205712679106964, "cls6_Classification": 0.2199305553747709, "cls6_Regression": 0.04138719761462976}, "epoch_time": 53622.7582423687}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_5.json

@@ -0,0 +1 @@
+{"train": {"Classification": 15.133819042362644, "Regression": 3.054872992772941, "MET": 4381.95359686758, "Sliced_Wasserstein_Loss": 2472.0649810471946, "Total": 18.188667628191777, "cls0_Classification": 7.596063778273003, "cls0_Regression": 0.0, "cls1_Classification": 1.4442247638709542, "cls1_Regression": 0.31361031255206817, "cls2_Classification": 1.575477985587537, "cls2_Regression": 4.2060938606448115, "cls3_Classification": 3.8002477142083557, "cls3_Regression": 29.90431176323572, "cls4_Classification": 3.3810041081767817, "cls4_Regression": 20.118552661723665, "cls5_Classification": 0.8077757106000333, "cls5_Regression": 0.5160948336683051, "cls6_Classification": 0.21218219311685133, "cls6_Regression": 0.04122732981363508}, "valid": {"Classification": 15.09622573725425, "Regression": 3.049987243835388, "MET": 4427.363878707098, "Sliced_Wasserstein_Loss": 2351.473008997001, "Total": 18.146224695934688, "cls0_Classification": 7.73505120688937, "cls0_Regression": 0.0, "cls1_Classification": 1.3740977942248418, "cls1_Regression": 0.2952244678206223, "cls2_Classification": 1.556745570486921, "cls2_Regression": 4.206630016556981, "cls3_Classification": 3.780003345239087, "cls3_Regression": 29.901655177440855, "cls4_Classification": 3.3507008080639786, "cls4_Regression": 20.13125312395868, "cls5_Classification": 0.7734659735817228, "cls5_Regression": 0.5103752037081389, "cls6_Classification": 0.20238484958138953, "cls6_Regression": 0.039524847410075545}, "epoch_time": 53709.221494436264}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_6.json

@@ -0,0 +1 @@
+{"train": {"Classification": 15.05975080184946, "Regression": 3.050915162765027, "MET": 4302.380139126089, "Sliced_Wasserstein_Loss": 2462.704211271712, "Total": 18.110637001707836, "cls0_Classification": 7.552141042196026, "cls0_Regression": 0.0, "cls1_Classification": 1.4260944724455367, "cls1_Regression": 0.30729873790066436, "cls2_Classification": 1.568396880336985, "cls2_Regression": 4.190179557004207, "cls3_Classification": 3.7984214239596783, "cls3_Regression": 29.892950493606033, "cls4_Classification": 3.3785578815553796, "cls4_Regression": 20.111691381680345, "cls5_Classification": 0.8027261579955847, "cls5_Regression": 0.508995952027013, "cls6_Classification": 0.20231808609858584, "cls6_Regression": 0.03994244405358984}, "valid": {"Classification": 15.03025553982006, "Regression": 3.047471155448184, "MET": 4199.804398533822, "Sliced_Wasserstein_Loss": 2408.510663112296, "Total": 18.07770326557814, "cls0_Classification": 7.516501010079973, "cls0_Regression": 0.0, "cls1_Classification": 1.433356524387704, "cls1_Regression": 0.2866287002639745, "cls2_Classification": 1.5437463228403032, "cls2_Regression": 4.184856349966678, "cls3_Classification": 3.814138451245418, "cls3_Regression": 29.875484213595467, "cls4_Classification": 3.388950102569977, "cls4_Regression": 20.122826245418192, "cls5_Classification": 0.7919665612764495, "cls5_Regression": 0.5052844933850175, "cls6_Classification": 0.20067208228767494, "cls6_Regression": 0.0396838952445539}, "epoch_time": 53527.57937669754}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_7.json

@@ -0,0 +1 @@
+{"train": {"Classification": 15.005305744157955, "Regression": 3.047662205294289, "MET": 4251.040779772567, "Sliced_Wasserstein_Loss": 2454.8403382346814, "Total": 18.052920762694214, "cls0_Classification": 7.5201399591785725, "cls0_Regression": 0.0, "cls1_Classification": 1.4122951118423792, "cls1_Regression": 0.30045564103048067, "cls2_Classification": 1.5632266760736453, "cls2_Regression": 4.179956952659641, "cls3_Classification": 3.7973950306160704, "cls3_Regression": 29.882997667347023, "cls4_Classification": 3.3773453445869954, "cls4_Regression": 20.106950576915068, "cls5_Classification": 0.7992619177479485, "cls5_Regression": 0.5042597845810597, "cls6_Classification": 0.19385703258424627, "cls6_Regression": 0.04022196966233156}, "valid": {"Classification": 14.98494512245918, "Regression": 3.045307488649617, "MET": 4336.255914695102, "Sliced_Wasserstein_Loss": 2470.9286904365213, "Total": 18.030263349716762, "cls0_Classification": 7.439019024908364, "cls0_Regression": 0.0, "cls1_Classification": 1.460192956514495, "cls1_Regression": 0.28850997014666985, "cls2_Classification": 1.5978667917985672, "cls2_Regression": 4.187001142848217, "cls3_Classification": 3.7926963798525493, "cls3_Regression": 29.8877874041986, "cls4_Classification": 3.3732906088595467, "cls4_Regression": 20.102260184105297, "cls5_Classification": 0.7884753747448767, "cls5_Regression": 0.49748712505987586, "cls6_Classification": 0.18429718983646076, "cls6_Regression": 0.03953910708149367}, "epoch_time": 53520.94456219673}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_8.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.961987649435581, "Regression": 3.0441124593868456, "MET": 4171.036447702753, "Sliced_Wasserstein_Loss": 2442.752363893864, "Total": 18.00608806806348, "cls0_Classification": 7.495415399675095, "cls0_Regression": 0.0, "cls1_Classification": 1.401329560128296, "cls1_Regression": 0.2930115638472633, "cls2_Classification": 1.5588417245490898, "cls2_Regression": 4.1723794101720335, "cls3_Classification": 3.796940085183488, "cls3_Regression": 29.871573915941184, "cls4_Classification": 3.3768705481734496, "cls4_Regression": 20.102545611696588, "cls5_Classification": 0.7965076754956888, "cls5_Regression": 0.5006904317620903, "cls6_Classification": 0.18616449271306285, "cls6_Regression": 0.03941630661678755}, "valid": {"Classification": 14.93864154656781, "Regression": 3.043543102819893, "MET": 4060.9506831056315, "Sliced_Wasserstein_Loss": 2455.8963678773744, "Total": 17.982203848717095, "cls0_Classification": 7.477601867085971, "cls0_Regression": 0.0, "cls1_Classification": 1.4040954447475842, "cls1_Regression": 0.281938389514693, "cls2_Classification": 1.5308484411446184, "cls2_Regression": 4.169689855985505, "cls3_Classification": 3.768482445955515, "cls3_Regression": 29.899270555648116, "cls4_Classification": 3.452811302482506, "cls4_Regression": 20.106226309980006, "cls5_Classification": 0.7831404831202099, "cls5_Regression": 0.49855264716448683, "cls6_Classification": 0.16842994067081807, "cls6_Regression": 0.039015429610214564}, "epoch_time": 53531.855256557465}

cms/2024_05_16_attn_model21M/pyg-cms_20240430_094836_751206/history/epoch_9.json

@@ -0,0 +1 @@
+{"train": {"Classification": 14.924325457158329, "Regression": 3.0432266500395717, "MET": 4131.878868663306, "Sliced_Wasserstein_Loss": 2441.514391635773, "Total": 17.967461313366936, "cls0_Classification": 7.474904454950639, "cls0_Regression": 0.0, "cls1_Classification": 1.3925459891906526, "cls1_Regression": 0.2923096608215833, "cls2_Classification": 1.5551009536281084, "cls2_Regression": 4.166108127317033, "cls3_Classification": 3.7959937466155704, "cls3_Regression": 29.85935091848211, "cls4_Classification": 3.3756472721497897, "cls4_Regression": 20.097993553963427, "cls5_Classification": 0.7942906300766444, "cls5_Regression": 0.4980677022587162, "cls6_Classification": 0.17899239024034658, "cls6_Regression": 0.03941535067815507}, "valid": {"Classification": 14.906765453182272, "Regression": 3.039978861212929, "MET": 4185.277240919693, "Sliced_Wasserstein_Loss": 2356.507497500833, "Total": 17.946753425941353, "cls0_Classification": 7.451920974258581, "cls0_Regression": 0.0, "cls1_Classification": 1.4142279721863547, "cls1_Regression": 0.2788756490349571, "cls2_Classification": 1.54328700329057, "cls2_Regression": 4.160146162216761, "cls3_Classification": 3.8120030953223925, "cls3_Regression": 29.858060334055313, "cls4_Classification": 3.37109375, "cls4_Regression": 20.104432637037654, "cls5_Classification": 0.7861675502697018, "cls5_Regression": 0.5026070004659905, "cls6_Classification": 0.1682531095591834, "cls6_Regression": 0.03847696875699637}, "epoch_time": 53528.248044490814}