meg HF staff commited on
Commit
f18e497
·
verified ·
1 Parent(s): 58b7eaf

Reading backend_model and experiment_name from parse_requests output.

Browse files
Files changed (1) hide show
  1. entrypoint.sh +3 -4
entrypoint.sh CHANGED
@@ -6,11 +6,10 @@ echo "Attempting to run."
6
  #if [[ $? = 0 ]]; then
7
 
8
  python /parse_requests.py > pending_backend_models.txt
9
- cat pending_backend_models.txt | while read backend_model; do
10
- # TODO: The set of experiments to run could also be defined in a text file.
11
- for experiment_name in 'text_classification'; do
12
  optimum-benchmark --config-name ${experiment_name} --config-dir /optimum-benchmark/examples/energy_star/ hydra.run.dir='/data/runs/${experiment_name}/${backend_model}/${now:%Y-%m-%d-%H-%M-%S}'
13
- done
14
  done
15
  echo "Finished"
16
  #python /pause_space.py
 
6
  #if [[ $? = 0 ]]; then
7
 
8
  python /parse_requests.py > pending_backend_models.txt
9
+ cat pending_backend_models.txt | while read line; do
10
+ backend_model=$(echo ${line} | awk -F',' '{print $1;}')
11
+ experiment_name=$(echo ${line} | awk -F',' '{print $2;}')
12
  optimum-benchmark --config-name ${experiment_name} --config-dir /optimum-benchmark/examples/energy_star/ hydra.run.dir='/data/runs/${experiment_name}/${backend_model}/${now:%Y-%m-%d-%H-%M-%S}'
 
13
  done
14
  echo "Finished"
15
  #python /pause_space.py