search

juntang-zhuang / LadderNet

139 stars 36 forks source link

MemoryError #3

Closed LJM-GitHub closed 5 years ago

LJM-GitHub commented 5 years ago

Traceback (most recent call last): File "retinaNN_training.py", line 122, in train_set = TrainDataset(patches_imgs_train[train_ind,...],patches_masks_train[train_ind,...]) MemoryError

What should I do? Thanks！

juntang-zhuang commented 5 years ago

Hi, Thanks for your feedback. Not sure what's the error. Try cleaning your GPU usage and memory, or try a smaller batchsize.

LJM-GitHub commented 5 years ago

Thanks,I try...

whikwon commented 5 years ago

For me, decreasing N_subimgs for CPU and batch_size for GPU worked.