Spaces:
Starting
on
A10G
Starting
on
A10G
Update entrypoint.sh
Browse files- entrypoint.sh +4 -4
entrypoint.sh
CHANGED
@@ -7,8 +7,8 @@ echo "Not checking h100 -- already know it's not there."
|
|
7 |
echo "Attempting to run."
|
8 |
#if [[ $? = 0 ]]; then
|
9 |
|
10 |
-
touch
|
11 |
-
touch
|
12 |
# For each line in the requests dataset....
|
13 |
python /parse_requests.py | while read -r line; do
|
14 |
# Read the name of the model and the experiment.
|
@@ -19,11 +19,11 @@ python /parse_requests.py | while read -r line; do
|
|
19 |
now=$(date +%Y-%m-%d-%H-%M-%S)
|
20 |
run_dir="./runs/${experiment_name}/${backend_model}/${now}"
|
21 |
mkdir -p "$run_dir"
|
22 |
-
echo "${experiment_name},${backend_model}" >>
|
23 |
|
24 |
# Let the benchmarking begin!
|
25 |
optimum-benchmark --config-name "${experiment_name}" --config-dir /optimum-benchmark/examples/energy_star/ backend.model="${backend_model}" backend.processor="${backend_model}" hydra.run.dir="${run_dir}" 2> "${run_dir}/error.log" ||
|
26 |
-
echo "${experiment_name},${backend_model}" >>
|
27 |
done
|
28 |
|
29 |
echo "Finished; updating requests dataset and results dataset."
|
|
|
7 |
echo "Attempting to run."
|
8 |
#if [[ $? = 0 ]]; then
|
9 |
|
10 |
+
touch /attempts.txt
|
11 |
+
touch /failed_attempts.txt
|
12 |
# For each line in the requests dataset....
|
13 |
python /parse_requests.py | while read -r line; do
|
14 |
# Read the name of the model and the experiment.
|
|
|
19 |
now=$(date +%Y-%m-%d-%H-%M-%S)
|
20 |
run_dir="./runs/${experiment_name}/${backend_model}/${now}"
|
21 |
mkdir -p "$run_dir"
|
22 |
+
echo "${experiment_name},${backend_model}" >> /attempts.txt
|
23 |
|
24 |
# Let the benchmarking begin!
|
25 |
optimum-benchmark --config-name "${experiment_name}" --config-dir /optimum-benchmark/examples/energy_star/ backend.model="${backend_model}" backend.processor="${backend_model}" hydra.run.dir="${run_dir}" 2> "${run_dir}/error.log" ||
|
26 |
+
echo "${experiment_name},${backend_model}" >> /failed_attempts.txt #(python /failed_run.py --run_dir "${run_dir}" --model_name "${backend_model}" && rm -rf $run_dir)
|
27 |
done
|
28 |
|
29 |
echo "Finished; updating requests dataset and results dataset."
|