What is the difference between `register_buffer` and `register_parameter` of `nn.Module`

10

@ptrblck probably another dumb question, but why wouldn’t I just use nn.Parameter for both my_tensor and my_param and just state ‘requires_grad=False’ for the first? How would that be different to the example in your post?

4 Likes