*** IndexError: index 1 is out of bounds for dimension 0 with size 1
复现Swin-Unet跑自己的数据的时候,出现的问题,主要原因就是数据是奇数,而batch_size开的偶数,因此[1]这个batch_size的位置有问题了,越界了,应该改成0。
*** IndexError: index 1 is out of bounds for dimension 0 with size 1
复现Swin-Unet跑自己的数据的时候,出现的问题,主要原因就是数据是奇数,而batch_size开的偶数,因此[1]这个batch_size的位置有问题了,越界了,应该改成0。