batch_size about R2AttU_Net?

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

2.68k stars 595 forks source link

batch_size about R2AttU_Net? #18

Closed zhaoxuyan closed 5 years ago

zhaoxuyan commented 5 years ago

I am using 1080ti I set batch_size = 2 of R2AttU_Net, it return cuda runtime error (2) : out of memory Can I set batch_size > 1 when i training R2AttU_Net?

LeeJunHyun commented 5 years ago

Hi, @zhaoxuyan . I think you should resize(reduce the size of) your input data, and fix data_loader.py code for pre-processing. (ex. zero padding for different size of images.)