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

Memory requirement. #5

Open
Tord-Zhang opened this issue Jun 24, 2019 · 2 comments
Open

Memory requirement. #5

Tord-Zhang opened this issue Jun 24, 2019 · 2 comments

Comments

@Tord-Zhang
Copy link

Tord-Zhang commented Jun 24, 2019

Impossible to run this model on single 1080ti GPU under the default setting in the code

@daitao
Copy link
Owner

daitao commented Jun 24, 2019

Impossible to run this model on single 1080ti GPU under the default setting in the code
The training is not a problem, since the model is trained by patches. For testing, it may be a problem if test images are too large. In practice, we can use 1x1 Conv or add pooling function to resize the resolution of feature input of SOCA for more efficiency.

@Tord-Zhang
Copy link
Author

Tord-Zhang commented Jun 25, 2019

However, the default patch size is set to 192. if the patch size is smaller than 192, the kernel size will be larger than the size of feature map in some SOCA module, which will bring some error when doing cuddn_conv. Besides, setting a smaller patch size, 48 for example (I am not sure if this patch size is small enough to run on single 1080ti), the performance would definitely be influenced.

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