pytoch 模型字典化后的参数更新

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'])

dd是在训练前就进行了参数的获取,那么在每一次进行训练的过程中,模型内部的参数都会变化,但是此时dd会变化吗?答案是会变化,也就是说,在模型的字典化中,字典里的值会随着模型的训练过程而变化的,虽然字典是在训练之前就定义好了。

发布了36 篇原创文章 · 获赞 11 · 访问量 6547

猜你喜欢

转载自blog.csdn.net/t20134297/article/details/103635711