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

关于显存问题 #54

Open

xk2016 opened this issue Jul 21, 2023 · 1 comment

xk2016 commented Jul 21, 2023

您好，
麻烦请问一下，感觉这套代码在训练过程中很占显存，想知道您在具体训练的过程中batchsize设置的多大？我目前采用2张2080Ti，batchsize设置成8的情况下，还是会出现显存溢出的情况，这个情况正常吗？期待您的恢复！Thanks!

祝好

Owner

JusperLee commented Nov 30, 2023

可以减少到batch=1

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