显存问题

[juewohu]

您好，我想问一下，这个上采样模块占用的显存非常大是正常的吗，我们大概训练了3000张1024*1024尺寸的图像，batch为2，使用三个模块，当第三个模块上采样到1024时，显存就爆炸了

[Linwei-Chen]

感谢关注！建议使用mmcv.ops.carafe作为backend的版本，对显存占用相对较少。