brats17 icon indicating copy to clipboard operation
brats17 copied to clipboard

OOM about training with brats17 data

Open Endousan opened this issue 5 years ago • 1 comments

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.

Endousan avatar Nov 01 '19 06:11 Endousan

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)

Dorin-D avatar Nov 18 '21 09:11 Dorin-D