sasha's picture
sasha HF staff
Update entrypoint.sh
2556dc3 verified
raw
history blame
751 Bytes
#!/bin/bash
echo "Not checking h100 -- already know it's not there."
#python /check_h100.py
echo "Attempting to run."
#if [[ $? = 0 ]]; then
python /parse_requests.py | while read line; do
IFS="," read backend_model experiment_name <<< $(echo ${line})
echo "Benchmarking Model: ${backend_model}, Task: ${experiment_name}"
optimum-benchmark --config-name ${experiment_name} --config-dir /optimum-benchmark/examples/energy_star/ backend.model=${backend_model} backend.processor=${backend_model} hydra.run.dir="./runs/${experiment_name}/${backend_model}/${now:%Y-%m-%d-%H-%M-%S}"
done
echo "Finished; uploading dataset results"
python /create_results.py ./runs
# Pausing space
echo "Pausing space."
python /pause_space.py
echo "Done."
#fi