Hello, I am trying to find some modules that have certain variables contained inside. I now can find these variables using
For instance, I may get a parameter and its name
module.layer1.0.conv1.weight, how can I find the corresponding
conv1 module it belongs to, in order to access other parameters inside this module? Is there any functions that I can use? Thanks in advance!
(I know that I can look for the module that is a instance of
model.module(), but problem is that it may contain other stuff like
nn.Sequential and other bigger blocks that contains
conv, making the code very complicated)