bubbliiiing / yolov5-v6.1-pytorch

这是一个yolov5-v6.1-pytorch的源码,可以用于训练自己的模型。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

关于训练时出错:fail to load key.请问该如何解决呢?

Liumj2022 opened this issue · comments

Successful Load Key: ['backbone.dark2.0.conv.weight', 'backbone.dark2.0.bn.weight', 'backbone.dark2.0.bn.bias', 'backbone.dark2.0.bn.running_mean', 'backbone.dark2.0.bn.running_var', 'backbone.dark2.0.bn.num_batches_tracked', 'backbone.dark2.1.cv1.conv.weight', 'backbone.dark2.1.cv1.bn.weight', 'backbone.dark2.1.cv1.bn.bias', 'backbone.dark2.1.cv1.bn.running_mean', 'backbone.dark2.1.cv1.bn.running_var', 'backbone.dark2.1.cv1.bn.num_batches_tracked', 'backbone.dark2.1.cv2.conv.weight', 'backbone.dark2.1.cv2.bn.weight ……
Successful Load Key Num: 308

Fail To Load Key: ['backbone.stem.conv.weight', 'backbone.stem.bn.weight', 'backbone.stem.bn.bias', 'backbone.stem.bn.running_mean', 'backbone.stem.bn.running_var', 'backbone.stem.bn.num_batches_tracked', 'backbone.dark5.1.cv2.conv.weight', 'backbone.dark5.1.cv2.bn.weight', 'backbone.dark5.1.cv2.bn.bias', 'backbone.dark5.1.cv2.bn.running_mean', 'backbone.dark5.1.cv2.bn.running_var', 'backbone.dark5.1.cv3.conv.weight', 'backbone.dark5.1.cv3.bn.weight', 'backbone.dark5.1.cv3.bn.bias', 'backbone.dark5.1.cv3.bn.runni ……
Fail To Load Key num: 40

环境是:
cuda11.8
python3.6
pytorch: 1.4.0