在训练模型时,Tensorflow GPU停止吸气

时间:2019-08-27 14:36:43

标签: python tensorflow faster-rcnn

我正在按照教程Youtube tutorial

训练用于自定义对象检测的模型

我正在尝试训练模型。

在执行步骤之前,它们会出现多个警告,例如-“分配器(GPU_0_bfc)内存不足,试图分配带有freed_by_count = 0的2.06GiB。调用方指出这不是失败,但可能表示性能有所提高。如果有更多的可用内存。” (请参见屏幕截图) warnings before step executions

然后,步骤就会像-

全局步骤1901:损失= 0.2125

..

..

,在某些步骤出现错误后,它的步骤执行停止(以下屏幕截图) steps get stoped with this errors

我正在使用faster_rcnn_inception_v2_pets.config模型进行训练 Tensorflow-1.14(最新) CUDA-10.1(最新) cucnn-10.1cuda-v7.63.30(lastest)

配置文件如下

我的文件是:-

B.__del__

0 个答案:

没有答案