PyTorch使用预训练模型

​​​​​

​

state_dict()

功能：获取模型当前的参数，以一个有序字典形式返回。

这个有序字典中，key 是各层参数名，value 就是参数。

load_state_dict(state_dict)

功能：将 state_dict 中的参数加载到当前网络，常用于 finetune。

​​python怎么使用预训练的模型_PyTorch使用预训练模型​​