ModuleList和Sequential在Pytorch中的区别

ModuleList和Sequential在Pytorch中的区别

一、分别简述

1、nn.Sequential

它里面的模块是按照顺序进行排列的，所以需要保证前一个模块的输出大小和下一个模块的输入大小一致。

2、nn.ModuleList

它是一个容器，可以储存不同的module，并将每个module添加到网络之中。类比认识的话，他就像是Python中常用到的list一样，就是一些放入和移除的操作。【特别注意：nn.ModuleList可以将module的所有Parameters放入网络之中】

class net_modlist(nn.Module):
    def __init__(self):
        super(net_modlist, self).__init__()
        self.modList