问题
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.
回答1:
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.
来源:https://stackoverflow.com/questions/47544051/when-should-i-use-nn-modulelist-and-when-should-i-use-nn-sequential