Skip to content
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.

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

question about out_channel #48

Open
shadowkiler opened this issue Nov 16, 2024 · 2 comments
Open

question about out_channel #48

shadowkiler opened this issue Nov 16, 2024 · 2 comments

Comments

@shadowkiler
Copy link

hello,thanks for sharing the code, i want use to DeformConvPack as encoder in 3D Unet, when i set out_channel = 32 in first layer,error an illegal memory access was encountered. and the out_channel = 16 and 24 in first layer. there are no errors. but the used memory=20G in A100 80G. Is it true that the higher the output channel, the more GPU memory it requires

@XinyiYing
Copy link
Owner

D3D requires more memory than shown. So just try to reduce batch size or patch size for less memory.

@shadowkiler
Copy link
Author

thanks very much,

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants