st_prediction_args.yml

entry: predict
batch_size: 64
search_method: beam_search
search_method.params:
  beam_size: 4
  length_penalty: -1
  maximum_decode_length: 180
metric: bleu

dataset.class: MultipleDataset
dataset.params:
  multiple_datasets:
    dev:
      dataset.class: AudioTFRecordDataset
      dataset.params:
        data_path: DATA_PATH/devtest/dev.en-TRG_LANG.tfrecords-00000-of-00001
        feature_key: audio
        transcript_key: translation
    tst-COM:
      dataset.class: AudioTFRecordDataset
      dataset.params:
        data_path: DATA_PATH/devtest/tst-COMMON.en-TRG_LANG.tfrecords-00000-of-00001
        feature_key: audio
        transcript_key: translation