image-restoration-sde icon indicating copy to clipboard operation
image-restoration-sde copied to clipboard

Published 20 hours ago • verified publisher icon Algolzw

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

Open LuoSSshuai opened this issue 5 months ago • 1 comments

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

LuoSSshuai avatar Sep 14 '24 11:09 LuoSSshuai