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
|