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_size改为1依旧OOM #37

Open

CDJ-happier opened this issue Apr 17, 2023 · 1 comment

CDJ-happier commented Apr 17, 2023 •

edited

Loading

我的GPU是2070，8G显存，batch_size改为1依旧爆显存，是本来对显存的要求大还是我设置得有问题？

Bin-ze commented Sep 27, 2023

1.把图像尺寸调小一点
2. 尝试将数据放置在CPu上
3. 是否数据集很大，如果是的话，考虑上面提到的两点应该可以解决你的问题

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