search

YubiaoYue / MedMamba

This is the official code repository for "MedMamba: Vision Mamba for Medical Image Classification"
297 stars 24 forks source link

GPU capacity #13

Open YamengGu opened 4 months ago

YamengGu commented 4 months ago

Hi,

I tried to train my data using this model. My NVIDIA GPU is 20GB. but the code kept saying the GPU is out of memory. i wonder if there is anyone training this model used which capacity GPU.

whaibetter commented 4 months ago

32g GPU is out of memory too;

jolexu commented 3 months ago

I have the same problem,and I tried to decrease the batch_size to 8 ,then it works ,but too slow.