ax.yaml 673 Bytes
Newer Older
Nikhilesh Bhatnagar's avatar
Nikhilesh Bhatnagar committed
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
# @package hydra.sweeper
_target_: hydra_plugins.hydra_ax_sweeper.ax_sweeper.AxSweeper
max_batch_size: null
ax_config:
  max_trials: 128
  early_stop:
    minimize: true
    max_epochs_without_improvement: 10
    epsilon: 0.025
  experiment:
    name: ${dataset.gen_subset}
    objective_name: wer
    minimize: true
    parameter_constraints: null
    outcome_constraints: null
    status_quo: null
  client:
    verbose_logging: false
    random_seed: null
  params:
    decoding.lmweight:
      type: range
      bounds: [0.0, 5.0]
    decoding.wordscore:
      type: range
      bounds: [-5.0, 5.0]
    decoding.silweight:
      type: range
      bounds: [ -8.0, 0.0 ]