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

Batch size #1

Closed

953951975 opened this issue Nov 30, 2023 · 1 comment

953951975 commented Nov 30, 2023

您好，请问为什么要将batchsize设置为2呢，我尝试将其设置为16，但是在第一个epoch运行到一半的时候报错：RuntimeError: CUDA error: an illegal memory access was encountered；但是将batchsize设置为2后又可以正常运行

Owner

caiyuanhao1998 commented Nov 30, 2023

hello，我把batch size设置为2是因为实验室机器显存大小的限制，我用的2080，所以设置小一点让它能够跑起来。我们的模型没有batch size，所以设置多大的batch size理论上是没啥太大影响的。

caiyuanhao1998 closed this as completed

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