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