Is there an equivalent to torch.nn.ModuleList in Rockpool ?

[pbonazzi]

I see I can easily convert a torch.nn.Module to TorchModule.
Is there an equivalent torch.nn.ModuleList/torch.nn.ConvTranspose2d in Rockpool ?

[DylanMuir]

Hi @pbonazzi , there's no equivalent to ModuleList, but the rockpool.nn.combinators.Sequential combinator is used to build feed-forward networks from a list of modules. Does that meet your needs? If not, what exactly do you need from ModuleList?

torch.nn.ConvTranspose2d should be able to be used natively in rockpool.

[pbonazzi]

Ok thank you Dylan. It does.

[DylanMuir]

Hi @pbonazzi , just wanted to check in — did you try using torch.nn.ConvTranspose2d in Rockpool? What was your experience?

[pbonazzi]

It did not raise any error when I used it :)

[DylanMuir]

Great!