first_net.load_state_dict(first_net_dict)
#first_net.eval()
dd = first_net.state_dict()
for data in dataloaders:
inputs,labels = data
x = first_net(inputs)
print(dd['bn1.running_mean'])first_net.load_state_dict(first_net_di
阅读背景：
pytoch 模型字典化后的参数更新
来源：互联网
分享到：
