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.