Hey!
Before building model, i was trying to use mateuszbuda brainsegmentation unet to see how it perfom on my multiple scelerosis dataset. I  have rtx 3070 with 8gb ram, and I am getting following error/warning: RuntimeError: CUDA out of memory. Tried to allocate 12.00 MiB (GPU 0; 7.77 GiB total capacity; 6.12 GiB already allocated; 35.75 MiB free; 6.22 GiB reserved in total by PyTorch)
Does anyone have an advice for me?
The link of the model is down below:
  
      
      github.com
  
  
    

mateuszbuda/brain-segmentation-pytorch 1


  U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI