Hi,
I would say the problem is you are  dropping  layers in a normal python list.
Can you use https://pytorch.org/docs/stable/nn.html?highlight=modulelist#torch.nn.ModuleList 15
nn.ModuleList instead?
A nn.Module is not aware of layers in a python list