taichi-nerfs icon indicating copy to clipboard operation
taichi-nerfs copied to clipboard

Published 20 hours ago • verified publisher icon taichi-dev

batch_size改为1依旧OOM

Open CDJ-happier opened this issue 1 year ago • 1 comments

我的GPU是2070，8G显存，batch_size改为1依旧爆显存，是本来对显存的要求大还是我设置得有问题？

CDJ-happier avatar Apr 17 '23 11:04 CDJ-happier

1.把图像尺寸调小一点 2. 尝试将数据放置在CPu上 3. 是否数据集很大，如果是的话，考虑上面提到的两点应该可以解决你的问题

Bin-ze avatar Sep 27 '23 14:09 Bin-ze