deeplabv3-plus-pytorch icon indicating copy to clipboard operation
deeplabv3-plus-pytorch copied to clipboard

模型权值载入问题

Open Voyagerlemon opened this issue 2 years ago • 1 comments

[1748380] (rank = 0, local_rank = 0) training... Gpu Device Count : 3 initialize network with normal type initialize network with normal type initialize network with normal type Load weights model_data/deeplab_mobilenetv2.pth.

Successful Load Key: ['backbone.features.0.0.weight', 'backbone.features.0.1.weight', 'backbone.features.0.1.bias', 'backbone.features.0.1.running_mean', 'backbone.features.0.1.running_var', 'backbone.features.0.1.num_batches_tracked', 'backbone.features.1.conv.0.weight', 'backbone.features.1.conv.1.weight', 'backbone.features.1.conv.1.bias', 'backbone.features.1.conv.1.running_mean', 'backbone.features.1.conv.1.running_var', 'backbone.features.1.conv.1.num_batches_tracked', 'backbone.features.1.conv.3.weight', 'bac …… Successful Load Key Num: 369

Fail To Load Key: ['cls_conv.weight', 'cls_conv.bias'] …… Fail To Load Key num: 2

温馨提示, head部分没有载入是正常现象, Backbone部分没有载入是错误的。 请问一下:“Fail To........”这两行显示的内容是网络有错误还是正常的?

Voyagerlemon avatar Jun 09 '23 07:06 Voyagerlemon

我也遇到这个问题,请问解决了吗

zhaoloulou avatar May 11 '24 02:05 zhaoloulou