search

microsoft / Recursive-Cascaded-Networks

[ICCV 2019] Recursive Cascaded Networks for Unsupervised Medical Image Registration
https://arxiv.org/abs/1907.12353
MIT License
361 stars 87 forks source link

GPU memery needed for training #41

Closed haoyayuzzz closed 3 years ago

haoyayuzzz commented 3 years ago

Thanks for sharing your code! I want to know the GPU memery needed during training. Because I always suffered the error about resource-exhaused. If I set the batchsize = 1 during traning in liver case, is a 12GB GPU enough to train VTN-10 model?

zsyzzsoft commented 3 years ago

I'm not sure as we were using our custom warping operation as explained here, and a 12GB GPU is enough in that case.

haoyayuzzz commented 3 years ago

Thanks for your reply. To my understanding, custom warping operation is not available for Ubuntu version, so I might try something else.