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

memory usage #109

Open

hjrPhoebus opened this issue May 6, 2024 · 0 comments

Assignees

hjrPhoebus commented May 6, 2024

您好，我想了解一下训练时候的显存占用。在训练时序attention的时候，对应的bs、帧数、卡数及单卡显存大概是什么样的对应关系呢？目前我能使用8张40G的A100显卡，不知道未来使用您的训练代码时大概可以开到多大的帧数？

AricGamma assigned subazinga

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment