File size: 1,257 Bytes
35d8d1f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 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 |