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.

Sign up for GitHub

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

Jump to bottom

speakers数量太大，导致最后的全连接层参数量巨大，然后就CUDA out of memory了 #63

Closed
quliulangle opened this issue Apr 29, 2024 · 5 comments
Closed

speakers数量太大，导致最后的全连接层参数量巨大，然后就CUDA out of memory了 #63

quliulangle opened this issue Apr 29, 2024 · 5 comments

Comments

@quliulangle
Copy link

speakers得有100万，batchsize已经设置的很小了，还是出错
@yeyupiaoling
Copy link
Owner

@quliulangle 这么大的说话人吗？有没有把speed_perturb_3_class设置为False

@yeyupiaoling
Copy link
Owner

在配置文件里面设置的。打印的模型结构里面，最后的分类层多少参数？

@quliulangle
Copy link
Author

image

@yeyupiaoling
Copy link
Owner

是有点离谱哦，显存多大，还不够？

@yeyupiaoling
Copy link
Owner

@quliulangle speed_perturb_3_class设置为False

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@yeyupiaoling @quliulangle