configs/text_classification.yaml

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