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

关于运行时占用显存的问题... #42

Open

dalaeyelids opened this issue Sep 26, 2024 · 0 comments

dalaeyelids commented Sep 26, 2024

作者您好，感谢您的工作。我在尝试用uctransnet跑自己的数据集时，发现其占用显存相当高（24g的显卡，跑512x512的RGB图像时batch_size设为8就会溢出，不得已只能设置为4），高于transunet和swinunet，请问这是uctransnet本身会占用更高显存吗？

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