推理的时候爆显存

[neverbedalao98]

我在用你们的预训练权重去推理cityscapes测试集的时候，sample per gpu设置为1，10G的显存还不够用，请问是可能是哪里出了问题？还是单纯的你们的模型需要很大的显存呢？（之前没有遇到过需要这么多显存的模型）

[Jiaqi-Chen-00]

We recommend to modify the test_cfg parameter in the config file, using dict(mode='slide', crop_size=(768, 768), stride=(512, 512)), to see if it yields any improvements.
If the GPU memory usage remains high, you can try reducing the crop_size and stride values further to save additional GPU memory.