infer.yaml 455 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
# @package _group_

defaults:
    - task: null
    - model: null

hydra:
  run:
    dir: ${common_eval.results_path}/${dataset.gen_subset}
  sweep:
    dir: ${common_eval.results_path}
    subdir: ${dataset.gen_subset}
common_eval:
  results_path: ${decoding.exp_dir}/decode/${decoding.decoder.name}
  path: ${decoding.exp_dir}/checkpoint_best.pt
  post_process: letter
generation:
  nbest: 1
  beam: 500
dataset:
  max_tokens: 1000000
  gen_subset: test