Adding pause at the end

configs/text_classification.yaml

@@ -1,46 +0,0 @@
-defaults:
-  - backend: pytorch # default backend
-  - launcher: process # default launcher
-  - benchmark: energy_star # default benchmark
-  - experiment # inheriting experiment schema
-  - _self_ # for hydra 1.1 compatibility
-  - override hydra/job_logging: colorlog # colorful logging
-  - override hydra/hydra_logging: colorlog # colorful logging
-
-experiment_name: text_classification
-
-launcher:
-  device_isolation: true
-  device_isolation_action: warn
-
-backend:
-  device: cuda
-  device_ids: 0
-  no_weights: true
-  model: lvwerra/distilbert-imdb
-  processor: lvwerra/distilbert-imdb
-  task: text-classification
-
-benchmark:
-  dataset_name: EnergyStarAI/text_classification
-  num_samples: 1000
-  text_column_name: text
-  truncation: True
-  input_shapes:
-    batch_size: 1
-
-# hydra/cli specific settings
-hydra:
-  run:
-    # where to store run results
-    dir: runs/${experiment_name}/${backend.model}/${now:%Y-%m-%d-%H-%M-%S}
-  sweep:
-    # where to store sweep results
-    dir: sweeps/${experiment_name}/${now:%Y-%m-%d-%H-%M-%S}
-  job:
-    # change working directory to the run directory
-    chdir: true
-    env_set:
-      # set environment variable OVERRIDE_BENCHMARKS to 1
-      # to not skip benchmarks that have been run before
-      OVERRIDE_BENCHMARKS: 1

entrypoint.sh

@@ -1,10 +1,10 @@
 #!/bin/bash
 
-echo "Checking h100"
+echo "Not checking h100 -- already know it's not there."
 #python /check_h100.py
 echo "Attempting to run."
 #if [[ $? = 0 ]]; then
-optimum-benchmark --config-name text_classification --config-dir /optimum-benchmark/examples/energy_star/ hydra.run.dir="/runs/" #$@ ##--config-dir /optimum-benchmark/examples/energy_star/ $@
+optimum-benchmark --config-name text_classification --config-dir /optimum-benchmark/examples/energy_star/ hydra.run.dir="/runs/"
 
-#python /pause_space.py
+python /pause_space.py
 #fi