defaults:
  - experiment # inheriting experiment schema
  - benchmark: inference
  - launcher: process
  - backend: pytorch
  - _self_ # for hydra 1.1 compatibility
  - override hydra/job_logging: colorlog # colorful logging
  - override hydra/hydra_logging: colorlog # colorful logging

experiment_name: pytorch_bert

launcher:
  device_isolation: true

benchmark:
  latency: true
  memory: true

backend:
  device: cuda
  device_ids: 0
  no_weights: true
  model: bert-base-uncased

# hydra/cli specific settings
hydra:
  run:
    # where to store run results
    dir: runs/${experiment_name}
  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