Skip to content

Conversation

@teamtee
Copy link
Contributor

@teamtee teamtee commented Apr 17, 2025

What does this PR do?

An example, aispeech_asr, and a large dataset, speech_dataset_large, have been added, and supporting multi-machine multi-GPU decoding.

Feature/Issue validation/testing

Please describe the tests that you ran to verify your changes and relevant result summary. Provide instructions so it can be reproduced.
Please also list any relevant details for your test configuration.

  • Add an example:aispeech_asr

This example is designed for large-scale industrial data training, suitable for datasets on the order of 100,000 hours. Its main features include:

  • Support for multi-task training: Designed to support tasks such as ASR and ST through a unified data format.

  • Dynamic prompt selection: Supports random selection from multiple prompts.

  • Iterative dataset: Uses an iterative dataset format to reduce startup time for large datasets.

  • Deepspeed training: Supports DeepSpeed training to significantly reduce memory usage.

  • Multi-machine multi-GPU inference: Supports distributed inference across multiple machines and GPUs to reduce evaluation time.

  • Dynamic frame batching: Dynamically combines frames based on audio size rather than using a fixed batch size, significantly reducing training and evaluation time (reduces training time by 3/4 for 100,000 hours of data).

  • Add an dataset for Dynamic frame batching

  • Support supporting multi-machine multi-GPU decoding.

Before submitting

Thanks for contributing 🎉!

@teamtee teamtee force-pushed the for-merge-aispeech-asr branch 2 times, most recently from b8016ef to 5ab1e81 Compare April 23, 2025 09:47
@ddlBoJack ddlBoJack merged commit 1f410bb into X-LANCE:main Apr 24, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants