加载预训练模型,出现问题
Unexpected key(s) in state_dict: “iteration”, “model_state_dict”, “optimizer_state_dict”, “scheduler”.
试试strict=False
for param in model.resnet.parameters():
param.requires_grad = False
model.resnet.load_state_dict(torch.load('./checkpoint_dir0/checkpoint1000.pt'),strict=False)
(笔记)