Error while loading weights

KeyError: 'unexpected key "module.features.conv0.weight" in state_dict'

I’m using densenet-201-kinetics.pth file with the densenet.py file from models folder.

net = densenet.densenet201(sample_size=64, sample_duration=30, num_classes=400)
pretrained_weights = torch.load(pretrained_path)
net.load_state_dict(pretrained_weights['state_dict'])

[kenshohara]

I could reproduce the error and solve it. I trained the model using GPUs. So, the model is wrapped by nn.DataParallel. You have to wrap the model using nn.DataParallel before load_state.

import torch
from torch import nn
import densenet
net = densenet.densenet201(sample_size=64, sample_duration=30, num_classes=400)

net = nn.DataParallel(net, device_ids=None)

pretrained_weights = torch.load('densenet-201-kinetics.pth')
net.load_state_dict(pretrained_weights['state_dict'])

[skrish13]

woohoo. thanks a lot! 😄