PyTorch Forums
What is the difference between `register_buffer` and `register_parameter` of `nn.Module`
pierrecurie
(pierrecurie)
October 9, 2019, 11:42pm
6
Do you want it in the state_dict?
show post in topic