今天小编就为大家分享一篇pytorch载入预训练模型后,实现训练指定层，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

1、有了已经训练好的模型参数，对这个模型的某些层做了改变，如何利用这些训练好的模型参数继续训练： pretrained_params = torch.load('Pretrained_Model'） model = The_New_Model(xxx) model.load_state_dict(pretrained_params.state_dict(), strict=False) strict=False 使得预训练模型参数中和新模型对应上的参数会被载入，对应不上或没有的参数被抛弃。 2