A5000 batchsize=1 CUDAoom

[stq-hydra]

跑DSText的baseline的时候，设置batchsize=1，在A5000单卡上出现CUDAoom
按照readme里面的描述 8张32G的V100能跑batchsize=16 那么24G单卡跑batchsize=1应该是不会出现这种情况的才对、
请问这种情况下还要调整哪些参数来降低显存占用

[weijiawu]

您好，可以贴一下报错的截图吗，batchsize为1应该不会爆显存，还有可以贴一下https://github.com/weijiawu/TransDETR/blob/main/configs/r50_TransDETR_train_DSText.sh这个文件的截图吗

我猜测可能你没改对bach size

[stq-hydra]

这是截图
早上进行测试的时候 将main.py中的dim_feedforward参数从1024改成了256之后可以勉强运行 但是两个小时之后仍然出现爆显存的情况

[weijiawu]

你把这里设置低一点，比如4，4,4 你这边的设置是需要处理9帧然后梯度反传

[stq-hydra]

设置为4 4 4之后，显存占用大约在16G，可以成功运行。
感谢！

[tangpan360]

设置为4 4 4之后，显存占用大约在16G，可以成功运行。 感谢！

您好，我看您在关注作者的工作，并且已经在一张A5000上完成对这个项目的复现。我最近也在关注作者的这项工作，但是对于复现和结果上有一点疑问，可以通过邮箱跟您请教一点相关的问题吗？如果方便的话这是我的邮箱：ptang@shu.edu.cn。期待您的回复，万分感谢！