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

知乎
神经网络模型保存下载model.state_dict()等用法和功能
神经网络模型保存下载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]}]}

保存模型

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

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

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

上一篇:Kotlin学习:第九章使用infix函数构建更可读的语法


下一篇:Codeforces Round #727 (Div. 2) C. Stable Groups