Caution while using nn.Parameter in Pytorch
Published in
2 min readOct 4, 2022
nn.Parameter fail to register in module.parameters()
While designing custom module with pytorch for my project, I required learnable parameters and I decided to rely on nn.Parameter(shape)
.
torch version: 1.8.1+cu111 & 1.12.1
Below is a mock example: