使用自己的数据集，运行test_MIF，报错RuntimeError: CUDA out of memory.各位大佬知道这么修改吗？

[1152067715]

RuntimeError: CUDA out of memory. Tried to allocate 3.61 GiB (GPU 0; 6.00 GiB total capacity; 2.45 GiB already allocated; 1.63 GiB free; 2.47 GiB reserved in total by PyTorch)
我觉得是应该修改Batch_size参数，但是在test阶段找不到如何修改，各位大佬能指点一下吗

[zhou770]

减小bach_size

[HaiyanggJiang]

显存爆了