Python与PyTorch系列(十四) 本文(2万字) | 解读Python中的register_buffer | 对比nn.Parameter register_buffer Tensor |

点击进入专栏：
《人工智能专栏》 Python与Python | 机器学习 | 深度学习 | 目标检测 | YOLOv5及其改进 | YOLOv8及其改进 | 关键知识点 | 各种工具教程

---

---

register_buffer

1.register_buffer( )的使用

回顾模型保存：torch.save(model.state_dict())，model.state_dict()是一个字典，里边存着我们模型各个部分的参数。
在model中，我们需要更新其中的参数，训练结束将参数保存下来。但在某些时候，我们可能希望模型中的某些参数参数不更新（从开始到结束均保持不变），但又希望参数保存下来（model.state_dict() ），这是我们就会用到 register_buffer() 。

随着例子边看边讲

例子1：使用类成员变量（类成员变量并不会在我们的model.state_dict()，即无法保存）

成员变量(self.tensor)在前向传播中用到，希望它也能保存下来，但他不在我们的state_dict中。