首页 > 其他分享 >PyTorch使用预训练模型

PyTorch使用预训练模型

时间:2023-02-22 22:01:05浏览次数:45  
标签:训练 模型 state PyTorch 参数 dict

​​

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

相关文章