关于采样阶段显存爆炸问题，训练正常

[LuoSSshuai]

我在deshadow文件夹训练正常，但是验证的时候就会显存爆炸，然后我更换显存更大的a800（80g显存）显卡也会爆。我在test也会发生显存爆炸的问题，如上图，请问有解决方法吗

[Algolzw]

你的validation图像是多大的呢，一直OOM的话建议先对图像downsample一下，deshadow后再upsample回去。如果在训练中也加入这个操作作为data augmentation性能应该不会有太大差距。