Hello.
I want to properly register a list of nn.Modules() that I structure as a list of list because they represent connections in a matrix. How is the best way to register a list of lists modules in pytorch?. As far as I understand from the doc, nn.ModuleList()
requires an interable containing modules, hence not sure if it is useful for my purposes.
I was thinking in creating a nn.Module
utility that contains each of the lists, and then create a nn.ModuleList
with a list of these modules.
Thanks in advance.