Skip to content

取消 OMP_NUM_THREADS 的注释有可能加速 encodec 训练 #41

@yt605155624

Description

@yt605155624

解开 launch.py 里面关于 OMP_NUM_THREADS 的注释可以加速训练,也能提高 GPU 利用率,因为默认会使用所有核心(对于核心数很多的机器如 A100),多核心之间的交互可能有耗时,如果觉得 1 太小,可以额外在 train.sh 前面控制(如使用 8),LibriTTS 的训练尚未测试

# if "OMP_NUM_THREADS" not in os.environ:

also see yangdongchao/SoundStorm#34

在该仓库中暂未验证

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions