coco数据集验证的时候报错

[chengfeng6s]

当我运行下面命令时， bash segm/dist_test.sh 4
--multiscale
--predict-dir start1.2_patch120_seg_deit_small_patch16_224_COCO_weaktr/seg_prob_ms
start1.2_patch120_seg_deit_small_patch16_224_COCO_weaktr/checkpoint.pth
coco

总是会在快结束的时候报错，如下

原本验证的时候用的是val_5000的验证集，可以运行完，但我改成了完整的验证集之后，运行到最后最后就会报错

这是什么原因呢，我查了一下有可能是因为内存不足，因为这个从头到尾总共要占用200多G内存，有什么办法可以减少内存占用啊

[chengfeng6s]

这个错误