Update clic/clusters/v1.9.0/README.md
Browse files
clic/clusters/v1.9.0/README.md
CHANGED
|
@@ -157,8 +157,7 @@ singularity exec -B /scratch/persistent --nv \
|
|
| 157 |
--env KERAS_BACKEND=torch \
|
| 158 |
$IMG python3 mlpf/pyg_pipeline.py --dataset clic --gpus 1 \
|
| 159 |
--data-dir /scratch/persistent/joosep/tensorflow_datasets --config parameters/pytorch/pyg-clic.yaml \
|
| 160 |
-
--test --make-plots --gpu-batch-multiplier 100 --load $WEIGHTS --dtype bfloat16 --prefetch-factor 10 --num-workers 8 --load $WEIGHTS --ntest 50000
|
| 161 |
-
~
|
| 162 |
```
|
| 163 |
|
| 164 |
</details>
|
|
|
|
| 157 |
--env KERAS_BACKEND=torch \
|
| 158 |
$IMG python3 mlpf/pyg_pipeline.py --dataset clic --gpus 1 \
|
| 159 |
--data-dir /scratch/persistent/joosep/tensorflow_datasets --config parameters/pytorch/pyg-clic.yaml \
|
| 160 |
+
--test --make-plots --gpu-batch-multiplier 100 --load $WEIGHTS --dtype bfloat16 --prefetch-factor 10 --num-workers 8 --load $WEIGHTS --ntest 50000
|
|
|
|
| 161 |
```
|
| 162 |
|
| 163 |
</details>
|