search

zju3dv / manhattan_sdf

Code for "Neural 3D Scene Reconstruction with the Manhattan-world Assumption" CVPR 2022 Oral
https://zju3dv.github.io/manhattan_sdf/
Other
491 stars 35 forks source link

为什么我运行strain_net.py代码的时候GPU内存极其容易爆满？ #23

Closed qinli-jian closed 1 year ago

qinli-jian commented 1 year ago

你好： 我在按照文件构建好项目，运行strain_net.py的时候，12G显存一下子就满了，我修改了batch_size也是容易爆满显存，请问你训练的机子都是显存很高的吗？ 谢谢！

ghy0324 commented 1 year ago

我们是用单张12G显卡训练的。你的pytorch版本是多少？建议使用1.6，参考：#8

czg-first commented 1 year ago

好像主要是由于query sdf这个过程比较占内存，其实训练过程不是很占显存。