1.batchsize不要太大,容易爆内存,一般就是4以下,比如2和4(2的n次方)
48就太大了(unet模型比较小,所以batch_size=48是可以的不会溢出内存,但是unet++模型比较大,所以batch_size=48就会溢出内存,实验时,我从48一直降到4才能将模型读入内存)
2.图片一般要裁剪成256*256或者是512*512又或者是224*224
1.batchsize不要太大,容易爆内存,一般就是4以下,比如2和4(2的n次方)
48就太大了(unet模型比较小,所以batch_size=48是可以的不会溢出内存,但是unet++模型比较大,所以batch_size=48就会溢出内存,实验时,我从48一直降到4才能将模型读入内存)
2.图片一般要裁剪成256*256或者是512*512又或者是224*224