神经网络模型保存下载model.state_dict()等用法和功能

知乎

总结：
model.parameters() 返回生成器
model.state_dict() 返回字典
字典格式{ 层的名（self.conv1 而不是nn.Conv2d）：对应的 参数矩阵 }
optimizer.state_dict() 返回字典
字典格式：其中包含两个元素 关注第二个：
param_groups对应的是一个只有一个字典的list
所以访问格式
optimizer.state_dict() [“param_groups”] [0][‘lr’]

{ “state”:{},“param_groups”：[{‘lr’: 0.001, ‘betas’: (0.9, 0.999), ‘eps’: 1e-08, ‘weight_decay’: 0, ‘amsgrad’: False, ‘params’: [0, 1]}]}

保存模型

知乎来源