gpt4 book ai didi

python - 训练自己的数据集。 Mask_RCNN资源耗尽: OOM when allocating

转载 作者:行者123 更新时间:2023-12-03 08:46:34 25 4
gpt4 key购买 nike

我正在尝试使用Mask_RCNN训练自己的数据集,但出现以下框架错误:
我已按照Mask R-CNN上的github tutorial进行对象检测。

我可以做些什么来减少训练数据集所需的内存吗?或者我该如何解决这个问题?

最佳答案

需要有关您的PC的更多信息。我认为可能是因为批处理大小或图像大小太大,因此需要更多的视频内存或内存。当我尝试在tensorflow-models/中使用mask-rcnn-resnet101训练自己的数据集时object_detection,GTX1080TI 11GB可以正常工作。(batch_size = 1,image_size = 1000 * 600)。

关于python - 训练自己的数据集。 Mask_RCNN资源耗尽: OOM when allocating,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52442789/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com