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

将bath_size设置为大于1时会报错 #44

Open

1404561326521 opened this issue Jul 1, 2024 · 0 comments

1404561326521 commented Jul 1, 2024

作者您好，感谢您的开源贡献，有一个问题需要请教一下，我的显卡为3090,24G显存，我在训练tgnet_bdl时将batch_size设置为1时训练一个epoch需要40分钟，对gpu的利用很小，我想充分利用gpu的作用，但是当我增大batch_size为2或者4时就会报错：

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