Training yolov3 Model[分配额外的工作空间_size=12.46 MB]中出错

Question

我在训练Yolov3模型时收到了这个错误消息，该模型具有max_batches=8000和27个filter和4个classess这是代码：/暗网探测器列车数据/obj.data cfg/yolov3_training.cfg/mydrive/yolov3/yolov3_training_list.weights-dont_show

这是错误：分配额外的工作空间_size=12.46 MB正在从/mydrive/yolov3/yolov3_training_last.weights加载权重…无法打开文件：/mydrive/olov3/yoov3_training_last.weights-

Answer 1

好的，所以你可以尝试解决几个问题：

首先，检查你是否分配了足够的内存，或者你的GPU是否内存不足。你有什么类型的GPU和VRAM？

如果内存不足，则必须减小.cfg文件中的批大小和细分。

其次，如果您使用CUDA，则需要导航到Darknet makefile并查找(应该是(第7行，这一行看起来像这样。

ARCH= -gencode arch=compute_20,code=[sm_20,sm_21]  

你必须在这里更换你的架构。darknet repo不包含它，但AlexeyAB的YoloV4 repo包含不同GPU的ARCH参数，仅与您的GPU匹配。如果你的GPU不在这里，你将不得不查找它。

# Tesla V100
# ARCH= -gencode arch=compute_70,code=[sm_70,compute_70]
# GeForce RTX 2080 Ti, RTX 2080, RTX 2070, Quadro RTX 8000, Quadro RTX 6000, Quadro RTX 5000, Tesla T4, XNOR Tensor Cores
# ARCH= -gencode arch=compute_75,code=[sm_75,compute_75]
# Jetson XAVIER
# ARCH= -gencode arch=compute_72,code=[sm_72,compute_72]
# GTX 1080, GTX 1070, GTX 1060, GTX 1050, GTX 1030, Titan Xp, Tesla P40, Tesla P4
# ARCH= -gencode arch=compute_61,code=sm_61 -gencode arch=compute_61,code=compute_61
# GP100/Tesla P100 - DGX-1
# ARCH= -gencode arch=compute_60,code=sm_60
# For Jetson TX1, Tegra X1, DRIVE CX, DRIVE PX - uncomment:
# ARCH= -gencode arch=compute_53,code=[sm_53,compute_53]
# For Jetson Tx2 or Drive-PX2 uncomment:
# ARCH= -gencode arch=compute_62,code=[sm_62,compute_62]

让我知道，如果这对你有效，我认为这些可能是手头的紧迫问题。