Joosep Pata
commited on
Commit
·
7eb3b5d
1
Parent(s):
3f61135
up
Browse files
clic/clusters/v2.0.0/README.md
CHANGED
|
@@ -64,7 +64,7 @@ git clone https://huggingface.co/jpata/particleflow models
|
|
| 64 |
|
| 65 |
## Training Details
|
| 66 |
Trained on 8x MI250X for 26 epochs over ~3 days.
|
| 67 |
-
The training was continued
|
| 68 |
|
| 69 |
### Training Data
|
| 70 |
The following datasets were used:
|
|
@@ -149,7 +149,6 @@ singularity exec -B /scratch/persistent --nv \
|
|
| 149 |
$IMG python3 mlpf/pyg_pipeline.py --dataset clic --gpus 1 \
|
| 150 |
--data-dir /scratch/persistent/joosep/tensorflow_datasets --config parameters/pytorch/pyg-clic.yaml \
|
| 151 |
--test --make-plots --gpu-batch-multiplier 100 --load $WEIGHTS --dtype bfloat16 --prefetch-factor 10 --num-workers 8 --load $WEIGHTS
|
| 152 |
-
~
|
| 153 |
```
|
| 154 |
|
| 155 |
## Citation
|
|
|
|
| 64 |
|
| 65 |
## Training Details
|
| 66 |
Trained on 8x MI250X for 26 epochs over ~3 days.
|
| 67 |
+
The training was continued from a checkpoint due to the 24h time limit.
|
| 68 |
|
| 69 |
### Training Data
|
| 70 |
The following datasets were used:
|
|
|
|
| 149 |
$IMG python3 mlpf/pyg_pipeline.py --dataset clic --gpus 1 \
|
| 150 |
--data-dir /scratch/persistent/joosep/tensorflow_datasets --config parameters/pytorch/pyg-clic.yaml \
|
| 151 |
--test --make-plots --gpu-batch-multiplier 100 --load $WEIGHTS --dtype bfloat16 --prefetch-factor 10 --num-workers 8 --load $WEIGHTS
|
|
|
|
| 152 |
```
|
| 153 |
|
| 154 |
## Citation
|