pytorch 状态字典:state_dict

pytorch 中的 state_dict 是一个简单的python的字典对象,将每一层与它的对应参数创建映射关系.(如model的每一层的weights及偏置等等)html (注意,只有那些参数能够训练的layer才会被保存到模型的state_dict中,如卷积层,线性层等等)python 优化器对象Optimizer也有一个state_dict,它包含了优化器的状态以及被使用的超参数(如lr,
相关文章
相关标签/搜索