search

med-air / 3DSAM-adapter

Holistic Adaptation of SAM from 2D to 3D for Promptable Medical Image Segmentation
134 stars 12 forks source link

Out of Memory #3

Open SillenceHu opened 1 year ago

SillenceHu commented 1 year ago

I use RTX 3060 to train model,but always out of memory,how can i do to fix code.

peterant330 commented 1 year ago

I use RTX 3060 to train model,but always out of memory,how can i do to fix code.

Hi,

Our code requires about 35G memory to run. We have tested with two 3090 or one A40.

WuJuli commented 1 year ago

hi, can you share your training code on 2 3090s?

I use RTX 3060 to train model,but always out of memory,how can i do to fix code.

Hi,

Our code requires about 35G memory to run. We have tested with two 3090 or one A40.