关于训练时出错: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