state_dict()
功能:获取模型当前的参数,以一个有序字典形式返回。
这个有序字典中,key 是各层参数名,value 就是参数。
load_state_dict(state_dict)
功能:将 state_dict 中的参数加载到当前网络,常用于 finetune。
python怎么使用预训练的模型_PyTorch使用预训练模型
标签:训练,模型,state,PyTorch,参数,dict From: https://blog.51cto.com/u_12074581/6079458