Hi everyone…
I noticed that we didn’t implement the Parameter() functionality in our Module class (ie, how weights are actually registered as parameters of the Linear layer ). The Pytorch source code is a bit confusing to me. Can someone either explain the method or the code to me?
I want to implement this feature without having to use nn.Linear( ie, by using a custom Linear Layer). Thanks a lot