module list

[wangjiawen2013]

Hi,
We can use nn.ModuleList([modules[i] for i in 0:range(num)]) in pytorch to construct a module list of various length(the modules in the list is not fixed and is determined by the range(num) ), how to achieve this in burn ?

[laggui]

You can use a vec to contain your modules, but the type must be known. If you want to store different types in the sequence, you can define an enum.

See also the discussion on nn.Sequential in #583.