wmt22-comet-da-eole-fp16

WMT22 COMET DA FP16 (EOLE)

This is Unbabel/wmt22-comet-da converted to EOLE format.

Model weights were dtype-cast to fp16 during EOLE conversion. This repository contains a format and dtype conversion for use with EOLE's native transformer_encoder_scorer implementation.

This is not an upstream unbabel-comet checkpoint layout. Use it with EOLE.

Model Details

Usage

Requires an EOLE version with native COMET scorer support.

Validation Metric

valid_metrics: ["EOLE-COMET"]
comet_model: eole-nlp/wmt22-comet-da-eole-fp16
comet_batch_size: 64

Direct Scoring

eole predict \
  --model_path eole-nlp/wmt22-comet-da-eole-fp16 \
  --src /path/to/src.txt \
  --tgt /path/to/mt.txt \
  --ref /path/to/ref.txt \
  --output /path/to/scores.txt \
  --with_score

By default, direct scoring writes one segment score per input line. To emit a single aggregate system score, add --score_level system:

eole predict \
  --model_path eole-nlp/wmt22-comet-da-eole-fp16 \
  --src /path/to/src.txt \
  --tgt /path/to/mt.txt \
  --ref /path/to/ref.txt \
  --output /path/to/system-score.txt \
  --with_score \
  --score_level system

score_level: system writes one numeric line containing the arithmetic mean of the segment scores, similar to Unbabel COMET's --only_system mode but using EOLE's score-file format.

Conversion

Converted with:

eole convert COMET \
  --model Unbabel/wmt22-comet-da \
  --dtype fp16 \
  --output wmt22-comet-da-eole-fp16

Parity

The fp32 EOLE conversion was parity-checked against the upstream Unbabel COMET runtime. On a 100-line WMT17 slice, native EOLE scoring matched the upstream baseline within floating point tolerance:

The fp16 artifacts are intended to provide smaller EOLE model files and fp16 runtime behavior. Run the EOLE COMET parity harness against these fp16 repositories if exact fp16 deltas are required.

Original Model and Attribution

This model is a conversion of Unbabel/wmt22-comet-da. The model weights were converted to EOLE format and dtype-cast to fp16.

This converted repository follows the upstream model license: Apache-2.0. Please refer to the original model card and license for training data, intended use, limitations, and citation details.

Unbabel COMET: https://github.com/Unbabel/COMET