- 因为声明的网络模型比保存的模型bn层多一个参数
num_batches_tracked
,所以跳过它
import torch, copy
net=yourNet()
state_dict = copy.deepcopy(net.state_dict())
keys=[]
for key in state_dict:
if(key
num_batches_tracked
,所以跳过它import torch, copy
net=yourNet()
state_dict = copy.deepcopy(net.state_dict())
keys=[]
for key in state_dict:
if(key