can't change batch_size

milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

GNU General Public License v3.0

9.3k stars 2.51k forks source link

Open wisezhou opened 8 months ago

wisezhou commented 8 months ago

i can't change batch_size from 1 to 2 ,the error is called RuntimeError: Trying to resize storage that is not resizable.

lipenghuai commented 3 months ago

same errors

pomeloyy commented 2 months ago

This error may occur if some images or labels in your dataset have different sizes.