模型训练/测试时的显存开销问题。

[1214635079]

作者你好，非常感谢你的代码。我在训练和测试模型时发现显存的开销非常大，当crop_size设为128时，在3090（24GB显存）上batch_size最高仅能设为2。因此在这种情况下，测试时也会超出显存，即使降低feature的通道数（nl_feas）也依然存在显存开销巨大的问题。我看到您在文章中是用1080Ti跑的实验，想请教一下我的设定是否存在问题？您训练或测试时通常的显存开销是多大？谢谢！

[Zlin0530]

作者你好，非常感谢你的代码。我在训练和测试模型时发现显存的开销非常大，当crop_size设为128时，在3090（24GB显存）上batch_size最高仅能设为2。因此在这种情况下，测试时也会超出显存，即使降低feature的通道数（nl_feas）也依然存在显存开销巨大的问题。我看到您在文章中是用1080Ti跑的实验，想请教一下我的设定是否存在问题？您训练或测试时通常的显存开销是多大？谢谢！

I also encountered the same problem, can you solve it?