pytorch 记录

pytorch 加载部分权重

pretrained_dict = torch.load('{0}/{1}'.format(opt.outf, opt.resume_posenet))
model_dict = model.state_dict()
#筛选条件
pretrained_dict =  {k: v for k, v in pretrained_dict.items() if k in model_dict}
model_dict.update(pretrained_dict)
model.load_state_dict(model_dict)

pytorch 固定权重

for p in self.parameters():
	p.requires_grad=False