Skip to main content

DiscoSeqSampler

CI codecov PyPI version Python version License: MIT

Distributed Coordinated Sequence Sampler - 一个高效的分布式序列采样框架。

背景

在当前的 AI 领域,无论是音频语音(Audio/Speech)还是图像视频(Image/Video)模型,都广泛使用 Transformer 架构。这类模型的计算量与序列长度高度相关,而在大规模数据集中,数据的长度分布往往非常广泛。为了实现高效的多 GPU 训练,必须对训练数据的序列长度进行精细准确的管理。

DiscoSeqSampler 正是为了解决这一关键问题而设计的分布式序列采样框架,它能够智能地协调和管理不同长度的序列数据,确保训练过程的高效性和稳定性。

特性

  • 🚀 高性能: 优化的分布式采样算法
  • 🔄 协调机制: 智能的序列协调和同步
  • 📊 可扩展: 支持大规模分布式部署
  • 🛠️ 易用性: 简洁的 API 设计
  • 🔧 可配置: 灵活的配置选项

安装

  • 项目仍在开发中,功能尚未完整验证

从 PyPI 安装

pip install discoss

从源码安装

git clone https://github.com/lifeiteng/DiscoSeqSampler.git
cd DiscoSeqSampler
pip install -e .

快速开始

import discoss

# TODO: 添加使用示例

开发

查看 DEVELOPMENT.md 获取详细的开发指南。

快速设置

# 克隆仓库
git clone https://github.com/lifeiteng/DiscoSeqSampler.git
cd DiscoSeqSampler

# 安装开发依赖
pip install -e .[dev]

# 设置 pre-commit 钩子
make setup-dev

运行测试

make test

贡献

欢迎贡献!请查看 DEVELOPMENT.md 了解如何设置开发环境。

许可证

本项目采用 MIT 许可证 - 查看 LICENSE 文件了解详情。

引用

如果您在研究中使用了 DiscoSeqSampler,请引用:

@software{discoss2024,
  title={DiscoSeqSampler: Distributed Coordinated Sequence Sampler},
  author={Feiteng Li},
  year={2025},
  url={https://github.com/lifeiteng/DiscoSeqSampler}
}

Metadata

Release files for discoss 0.1.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for discoss 0.1.1
File Size Uploaded Provenance Actions
discoss-0.1.1.tar.gz 20.1 kB Without provenance Details for discoss-0.1.1.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for discoss 0.1.1
File Interpreter ABI Platform Actions
discoss-0.1.1-py3-none-any.whl Python 3 none any Details for discoss-0.1.1-py3-none-any.whl

Total release size: 36.6 kB

Release files / discoss-0.1.1.tar.gz

Download URL discoss-0.1.1.tar.gz
Size 20.1 kB
Tags Source
SHA-256 checksum
How to use checksums
f049c242687fcfb675afbc577d52a06d33c3af76a335c541af597499caa59b98
BLAKE2b-256 checksum
How to use checksums
de8eb2a8cb82fa44f5dbb15230d99337d7ce57e5b4c7304365c559652408b34b
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.13.5

Release files / discoss-0.1.1-py3-none-any.whl

Download URL discoss-0.1.1-py3-none-any.whl
Size 16.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
3b8cd95c48025734ca5d4ae996946ad842b18e18eb300f0a79deed86271caa35
BLAKE2b-256 checksum
How to use checksums
ee87ddb3208e6c620532282113a9d1eb1c63125dd46f4eac7c1d9e76852c4bf4
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.13.5

Security for discoss 0.1.1

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

0.1.1 This release

2 release files No provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page