Repository navigation
Add flores200 benchmark - #1259
Merged
bxyu-nvidia merged 1 commit intoMay 7, 2026
Merged
bxyu-nvidia merged 1 commit into
bxyu-nvidia merged 1 commit into
Conversation
Adds the FLORES-200 multilingual translation benchmark (devtest split, default 6 languages = en/de/es/fr/it/ja, 30 directed pairs Γ 1012 = 30360 rows) on top of the existing wmt_translation resource server. Prepare-output is byte-identical to NeMo-Skills' flores200/prepare.py (md5 8e0c08feb62f336b4c5fe61ccdadd9bc); prompt template is character-identical to multilingual/segment-translation.yaml. 5 example rollouts captured against a hosted OpenAI-compatible endpoint with all rewards in [0,1] and non-empty translations. Signed-off-by: gwarmstrong <georgea@nvidia.com>
bxyu-nvidia
approved these changes
May 7, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Add flores200 benchmark
Migrates the
flores200benchmark from NeMo Skills into Gym on top of theexisting
wmt_translationresource server (introduced forwmt24ppin #1199).The same deterministic corpus-BLEU + optional xCOMET-XXL verifier scores all
30 directed pairs of FLORES+ devtest's default six-language set (
en,de,es,fr,it,ja).Includes
benchmarks/flores200/β benchmark config, prepare.py, prompt template, sample rolloutsopenlanguagedata/flores_plus,devtestsplit, six configs(
eng_Latn,deu_Latn,spa_Latn,fra_Latn,ita_Latn,jpn_Jpan)byte-equivalent to NeMo-Skills'
flores200/prepare.pyoutput under thesame defaults β md5
8e0c08feb62f336b4c5fe61ccdadd9bc)prompt_configat rollout time βprompts/default.yaml, character-identical to NeMo-Skills'multilingual/segment-translation.yamlwmt_translationresource server via_inherit_from; per-row fields (text,translation,source_language,target_language,source_lang_name,target_lang_name) match theserver's
WmtTranslationRunRequestschemaprepare.pycarries the same xCOMET-XXL + xlm-roberta-xxl prefetchhelper as
wmt24pp/prepare.py, so first-job COMET startup stays fullyoffline
data/example_rollouts.jsonlβ 5 rollouts collected against a hostedOpenAI-compatible endpoint (Nemotron-Nano 9B v2), all with
reward β [0, 1]and non-empty translationsValidated against NeMo Skills
prepare.pyoutput is byte-identical to NeMo-Skills' reference under thesame default arguments (split=
devtest, languages=[en, de, es, fr, it, ja]):The shared
wmt_translationresource server (landed in the parent PR) isreused unchanged: deterministic corpus-BLEU per language pair plus the
xx->xx,<src>->xx,xx->{tgt}aggregations, optionally augmented byxCOMET-XXL when
compute_comet: trueis set on the server. Verificationparity with NeMo-Skills'
TranslationMetricswas already established forwmt24ppand applies here unchanged β the verifier reads the same rowfields and emits the same key set.
A 5-row smoke run via
ng_e2e_collect_rolloutsagainst a hostedOpenAI-compatible endpoint produced 5/5 non-empty translations with sentence
BLEU spread across
[11.7, 42.0], confirming the prompt + per-row schemaplumb cleanly through the existing
wmt_translation_simple_agentβwmt_translationβvllm_modelchain. (Heavy parity at production scale isinherited from the parent
wmt24ppPR's full-cluster comparison; no modelchanges here.)