时间戳

首页 > 解决方案 > Yolov4 训练中的问题与备份权重

google-colaboratory - Yolov4 训练中的问题与备份权重

问题描述

我正在尝试使用 colab 中已经保存的权重来训练 yolov4。但是训练过程在加载权重后突然停止。下面是日志，在创建 6 个永久 cpu-threads后，执行停止：

[yolo] params: iou loss: ciou (4), iou_norm: 0.07, cls_norm: 1.00, scale_x_y: 1.05
  nms_kind: greedynms (1), beta = 0.600000 
  Total BFLOPS 59.563 
  avg_outputs = 489778 
  Allocate additional workspace_size = 52.43 MB 
  Loading weights from 
 /content/drive/MyDrive/YOLOv4_weight/backup/yolov4_custom_train_final.weights...
 seen 64, trained: 192 K-images (3 Kilo-batches_64) 
Done! Loaded 162 layers from weights-file 
Learning Rate: 0.01, Momentum: 0.949, Decay: 0.0005
Detection layer: 139 - type = 27 
Detection layer: 150 - type = 27 
Detection layer: 161 - type = 27 
Saving weights to backup/yolov4_custom_train_final.weights
Create 6 permanent cpu-threads

我尝试使用的权重是在 3000 个 epoch 之后生成的，并且使用 yolov4.conv.137 作为初始权重数据；由于准确度很低，我想用保存的权重运行 5000 个 epoch。

标签： google-colaboratoryyolo

解决方案

推荐阅读