Adding text_classification.yaml

configs/text_classification.yaml

@@ -0,0 +1,46 @@
+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