24G显存evalaute时候batch size是1不够嘛
Opened this issue · 2 comments
andypinxinliu commented
我在运行的时候，发现render那一块非常占显存，在4090上做evaluation的时候batch size是1的时候都会cuda out of memory，我想问一下这个是代码有些地方没有优化还是说本身确实需要很大的计算量，如果需要很大的计算量的话，大概是多少的memory才比较合适啊
SeanChenxy commented
Hi, 如果不使用2D超分模块，确实不够，减小 chunk 可以避免evaluation的OOM的问题
andypinxinliu commented
我还有一个小问题，default里的patch是0.125，那么这个patch size对于模型的训练有什么样的影响呢，调大了是不是会减少模型的效果啊