search

taigw / brats17

Brain tumor segmentation for MICCAI 2017 BraTS challenge
BSD 3-Clause "New" or "Revised" License
321 stars 130 forks source link

OOM about training with brats17 data #39

Open Endousan opened 4 years ago

Endousan commented 4 years ago

hello, I have some problem about this code, please help me. I use the GPU: v100 16G. but when I try to train the net with brats17 data by your code, I get OOM. I only used 70% data of brats17. from the code, It looks like it load all the data into it. I hope to know some about dataloading when your training with brats17.

Dorin-D commented 2 years ago

The pipeline loads the entire dataset into RAM. With the BraTS2020 dataset, 32RAM was not enough. 64RAM did the trick (296 subjects, 80% of the provided training set)