state_dict()
功能:获取模型当前的参数,以一个有序字典形式返回。
这个有序字典中,key 是各层参数名,value 就是参数。
load_state_dict(state_dict)
功能:将 state_dict 中的参数加载到当前网络,常用于 finetune。
state_dict()
功能:获取模型当前的参数,以一个有序字典形式返回。
这个有序字典中,key 是各层参数名,value 就是参数。
load_state_dict(state_dict)
功能:将 state_dict 中的参数加载到当前网络,常用于 finetune。