clarify training details
Browse files
cms/2024_04_05/pyg-cms_20240324_235743_208080/README.md
CHANGED
|
@@ -6,7 +6,7 @@ This model reconstructs particles in a detector, based on the tracks and calorim
|
|
| 6 |
|
| 7 |
### Model Description
|
| 8 |
|
| 9 |
-
- **Developed by:** Joosep Pata, Eric Wulff,
|
| 10 |
- **Model type:** full attention
|
| 11 |
- **License:** Apache License
|
| 12 |
|
|
@@ -35,6 +35,9 @@ It should not be used outside of the CMS collaboration.
|
|
| 35 |
|
| 36 |
## Training Details
|
| 37 |
|
|
|
|
|
|
|
|
|
|
| 38 |
### Training Data
|
| 39 |
|
| 40 |
Trained on 400k events from `cms_pf_ttbar`, version `v1.7.1`.
|
|
@@ -43,6 +46,3 @@ https://github.com/jpata/particleflow/blob/v1.7.0/mlpf/heptfds/cms_pf/ttbar.py
|
|
| 43 |
## Model Card Contact
|
| 44 |
|
| 45 |
Joosep Pata, [email protected]
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
|
|
|
| 6 |
|
| 7 |
### Model Description
|
| 8 |
|
| 9 |
+
- **Developed by:** Joosep Pata, Farouk Mokhtar, Eric Wulff, Shivam Raj, Javier Duarte
|
| 10 |
- **Model type:** full attention
|
| 11 |
- **License:** Apache License
|
| 12 |
|
|
|
|
| 35 |
|
| 36 |
## Training Details
|
| 37 |
|
| 38 |
+
Trained for 32 epochs on 1x A100 80GB for approximately 6 days.
|
| 39 |
+
The training was done with bfloat16.
|
| 40 |
+
|
| 41 |
### Training Data
|
| 42 |
|
| 43 |
Trained on 400k events from `cms_pf_ttbar`, version `v1.7.1`.
|
|
|
|
| 46 |
## Model Card Contact
|
| 47 |
|
| 48 |
Joosep Pata, [email protected]
|
|
|
|
|
|
|
|
|