tensorflow更快RCNN对象检测api的Google colab内存允许错误

tensorflow更快RCNN对象检测api的Google colab内存允许错误,tensorflow,google-colaboratory,object-detection-api,faster-rcnn,Tensorflow,Google Colaboratory,Object Detection Api,Faster Rcnn,我在自定义数据集上的GoogleColab中运行更快的RCNN,但它给出了与内存相关的错误,我无法通过。我在TPU上设置了运行时(也在GPU上尝试过,并在pipeline_file.config中设置batch_size=1) 完全错误是 部分错误是: Errors may have originated from an input operation. Input Source operations connected to node MultiLevelMatMulCropAndRe

我在自定义数据集上的GoogleColab中运行更快的RCNN,但它给出了与内存相关的错误,我无法通过。我在TPU上设置了运行时(也在GPU上尝试过,并在pipeline_file.config中设置batch_size=1)

完全错误是

部分错误是:

Errors may have originated from an input operation.
Input Source operations connected to node     MultiLevelMatMulCropAndResize/MultiLevelRoIAlign/GatherV2_1:
MultiLevelMatMulCropAndResize/MultiLevelRoIAlign/mul_14 (defined at  /local/lib/python3.7/dist-packages/object_detection/utils/spatial_transform_ops.py:273)

Function call stack:
_dummy_computation_fn
  • 当我在GPU上训练高效-d0模型时,它已成功训练,但训练更快的RCNN不起作用
任何形式的帮助都将不胜感激