Skip to content

[s2s] 1 GPU test for run_distributed_eval #8297

Description

@sshleifer

Add test coverage for run_distributed_eval.py that can run on 1 GPU.
The command:

python -m torch.distributed.launch --nproc_per_node=1 run_distributed_eval.py --model_name Helsinki-NLP/opus-mt-en-ro --save_dir opus_wmt_en_ro_gens --data_dir wmt_en_ro

works on 1 GPU.

After adding test coverage, we could try to improve API consistency between run_distributed_eval.py and run_eval.py .

Metadata

Metadata

Assignees

Labels

Help wantedExtra attention is needed, help appreciated

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions