I am new to Pytorch and one thing that I don't quite understand is the usage of nn.ModuleList
and nn.Sequential
. Can I know when I should use one over the other? Thanks.
nn.ModuleList
does not have a forward
method, but nn.Sequential
does have one. So you can wrap several modules in nn.Sequential
and run it on the input.
nn.ModuleList
is just a Python list (though it's useful since the parameters can be discovered and trained via an optimizer). While nn.Sequential
is a module that sequentially runs the component on the input.
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With