pytorch载入预训练模型后,只想训练个别层怎么办?
1、有了已经训练好的模型参数,对这个模型的某些层做了改变,如何利用这些训练好的模型参数继续训练:pretrained_params = torch.load('Pretrained_Model')model = The_New_Model(xxx)model.load_state_dict(pretrained_params.state_dict(), strict=False)str...
原创
2018-04-17 09:33:18 ·
13714 阅读 ·
1 评论