Author, how big is your GPU display for training this model? - Githubissues

CSCYQJ / LOCATION-SENSITIVE-LOCAL-PROTOTYPE-NETWORK

For ISBI 2021 paper "A LOCATION-SENSITIVE LOCAL PROTOTYPE NETWORK FOR FEW-SHOT MEDICAL IMAGE SEGMENTATION"

9 stars 1 forks source link

Author, how big is your GPU display for training this model? #3

Closed smallkaka closed 1 year ago

smallkaka commented 1 year ago

Author, how big is your GPU display for training this model?

CSCYQJ commented 1 year ago

I trained my model on a single NVIDIA TITAN RTX GPU of 24 GB memory.

smallkaka commented 1 year ago

Thank you for your reply. My single gpu is only 12GB. Can I try to change 512512 to 256256?