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

请问下这个训练为什么这么慢，每个batch训练时占用的显存特别小 #26

Open
estimate123 opened this issue Dec 8, 2020 · 2 comments
Open

请问下这个训练为什么这么慢，每个batch训练时占用的显存特别小 #26

estimate123 opened this issue Dec 8, 2020 · 2 comments

Comments

@estimate123
Copy link

你好，请问下我跑这个代码时为什么训练速度特别慢，感觉是一个batch一个batch跑的，显存只占用了306MB，没有并行跑起来，跑一个epoch可能就得跑好几个小时，请问下这是为什么？您训练时遇到这样的问题了吗？十分感谢
@WoNiuHu
Copy link

WoNiuHu commented Jan 12, 2021

请问这个问题解决了吗？我也碰到这个问题，求指导下

你好，请问下我跑这个代码时为什么训练速度特别慢，感觉是一个batch一个batch跑的，显存只占用了306MB，没有并行跑起来，跑一个epoch可能就得跑好几个小时，请问下这是为什么？您训练时遇到这样的问题了吗？十分感谢

@stanli124
Copy link

pytorch那个版本是cpu跑的，tf你可以注意一下

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
3 participants
@WoNiuHu @stanli124 @estimate123