meg HF staff commited on
Commit
807b73e
·
verified ·
1 Parent(s): 4882a1e

Starting ability to parse the requests files.

Browse files
Files changed (1) hide show
  1. entrypoint.sh +8 -1
entrypoint.sh CHANGED
@@ -4,7 +4,14 @@ echo "Not checking h100 -- already know it's not there."
4
  #python /check_h100.py
5
  echo "Attempting to run."
6
  #if [[ $? = 0 ]]; then
7
- optimum-benchmark --config-name text_classification --config-dir /optimum-benchmark/examples/energy_star/ hydra.run.dir='/data/runs/${experiment_name}/${backend.model}/${now:%Y-%m-%d-%H-%M-%S}'
 
 
 
 
 
 
 
8
  echo "Finished"
9
  #python /pause_space.py
10
  #fi
 
4
  #python /check_h100.py
5
  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
17
  #fi