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

YoutubeDNN训练很慢, YoutubeDNN training so slow #88

Open
johndkl opened this issue Sep 30, 2022 · 0 comments
Open

YoutubeDNN训练很慢, YoutubeDNN training so slow #88

johndkl opened this issue Sep 30, 2022 · 0 comments
Labels
question Further information is requested

Comments

@johndkl
Copy link

johndkl commented Sep 30, 2022

我的youtubednn训练很慢, 1024一个batch要花1分钟来训练, 也就是说500万行数据要2,3天, 同时显存占用率很高但是使用率低。
可能是数据读取比较慢但是测过数据读取时间应该不是这方面原因, 同样的数据读取格式DSSM只需要半小时, 差几十倍速度。

会不会是和我的mid size有关, 我的mid vocabulary size是1000万, 负采样num_sampled是500。 理论上SampledSoftmaxLoss的话和总的mid vocabulary size没有太大关系吧, 我也试过num_sampled=5,时间上没有变化。
不知道有人碰到过相似情况吗,谢谢。

@johndkl johndkl added the question Further information is requested label Sep 30, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
question Further information is requested
Projects
None yet
Development

No branches or pull requests

1 participant