Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
yt605155624 opened this issue Aug 11, 2023 · 0 comments
Open

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

yt605155624 opened this issue Aug 11, 2023 · 0 comments

Comments

@yt605155624
Copy link
Contributor

yt605155624 commented Aug 11, 2023

解开 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

在该仓库中暂未验证

@yt605155624 yt605155624 changed the title 取消 OMP_NUM_THREADS 的注释 取消 OMP_NUM_THREADS 的注释有可能加速 encodec 训练 Aug 11, 2023
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

No branches or pull requests

1 participant