Prediction configuration

An example configuration file is provided in config/predict.yaml:

use_gpu: true

run:
    output_dir: predictions
    predict_on_data_partition: test

defaults:
    - data: wmt20.qe.en_de

system:
    load: best_model.torch

    batch_size: 16
    num_data_workers: 0

    model:
        outputs:
            word_level:
                target: true
                gaps: true
                source: true
            sentence_level:
                hter: true

Configuration class

Full API reference: kiwi.lib.predict.Configuration

Training configuration Evaluation configuration