How to create a list of modulelists

autograd
1

Hi,
Is it ok to create a list of module list?
If for example I want to have a few Conv1d in a layer and than another layer with different Conv1d, in each layer I need to do a different manipulation on the output depending on the layer, what is the correct way to build this “list” of modulelists?

This way:

    class test(nn.Module):
        def __init__(...):
            self.modulelists = []
            for i in range(4):
                self.modulelists.append(nn.ModuleList([nn.Conv1d(10, 10, kernel_size=5) for _ in range(5)]))

or this way:

    class test(nn.Module):
        def __init__(...):
            self.modulelists = nn.ModuleList()
            for i in range(4):
                self.modulelists.append(nn.ModuleList([nn.Conv1d(10, 10, kernel_size=5) for _ in range(5)]))

Thanks