mask rcnn 推理和验证(eval)时显存占用比训练时的大

[DreamMaker777]

使用的是robot_grab的例程，数据的transforms如下图：

当设置eval_dataset时，在eval时候显存耗尽，但是训练时显存占用正常(batch size=4)：

训练时显存占用：

开始验证时报错：

同样，推理时也显存耗尽：

原始图像大小有两种，3072x2304和1280x720。推理时用1280x720分辨率不是一开始耗尽，而是推理大约十张耗尽；3072x2304分辨率直接耗尽。请问有什么解决方法吗？