hrnet-pytorch icon indicating copy to clipboard operation
hrnet-pytorch copied to clipboard

HRNetW48无法加载权重

Open Cochary opened this issue 2 years ago • 4 comments

使用提供的W48imagenet预训练权重,但backbone部分全部无法成功加载

Cochary avatar Sep 12 '22 03:09 Cochary

pretrained = True

bubbliiiing avatar Sep 13 '22 14:09 bubbliiiing

pretrained = True

train.py设置: backbone = "hrnetv2_w48" pretrained = True model_path = "model_data/hrnetv2_w48_imagenet_pretrained.pth"

加载失败: Load weights model_data/hrnetv2_w48_imagenet_pretrained.pth.

Successful Load Key: [] …… Successful Load Key Num: 0

Fail To Load Key: ['conv1.weight', 'bn1.weight', 'bn1.bias', 'bn1.running_mean', 'bn1.running_var', 'bn1.num_batches_tracked', 'conv2.weight', 'bn2.weight', 'bn2.bias', 'bn2.running_mean', 'bn2.running_var', 'bn2.num_batches_tracked', 'layer1.0.conv1.weight', 'layer1.0.bn1.weight', 'layer1.0.bn1.bias', 'layer1.0.bn1.running_mean', 'layer1.0.bn1.running_var', 'layer1.0.bn1.num_batches_tracked', 'layer1.0.conv2.weight', 'layer1.0.bn2.weight', 'layer1.0.bn2.bias', 'layer1.0.bn2.running_mean', 'layer1.0.bn2.running_v …… Fail To Load Key num: 1956

Cochary avatar Sep 13 '22 15:09 Cochary

……好好看看注释呢

bubbliiiing avatar Sep 14 '22 14:09 bubbliiiing

我的也是这样,解决了吗?

ruarua6666 avatar Feb 09 '23 14:02 ruarua6666