deeplabv3-plus-pytorch
deeplabv3-plus-pytorch copied to clipboard
模型权值载入问题
[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........”这两行显示的内容是网络有错误还是正常的?
我也遇到这个问题,请问解决了吗