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

Validating的时候特别慢 #14

Open
shiyao1999 opened this issue Nov 19, 2023 · 3 comments
Open

Validating的时候特别慢 #14

shiyao1999 opened this issue Nov 19, 2023 · 3 comments

Comments

@shiyao1999
Copy link

shiyao1999 commented Nov 19, 2023 •
edited
Loading

我用它来训练一个离线中文文字识别，训练集大约5w张，验证集大约1w张，用4块A5000训练，training的时候大约30分钟1个epoch，但是Validaing的时候很慢，差不多6个小时才能valida一次，这是什么原因。
捕获

pytorch_lightning==1.3.5
torch==1.8.1
@shiyao1999
Copy link
Author

已经解决了，验证的时候束宽为10，需要太多的计算资源，调小一点就好了

@121649982
Copy link

已经解决了，验证的时候束宽为10，需要太多的计算资源，调小一点就好了

在哪里设置的？

@shiyao1999
Copy link
Author

已经解决了，验证的时候束宽为10，需要太多的计算资源，调小一点就好了

在哪里设置的？
把beam_size调小

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
@121649982 @shiyao1999