I would be a bit careful using this code:



 Flock1:

test_block = torch.nn.Sequential(*list(test_model.children()))[:2]


as it will create an nn.Sequential module using the child modules in a sequential order.
In your initial code it seems that
(conv_final): Conv2d(8, 1, kernel_size=(1, 1), stride=(1, 1))

is initialized as the first layer, but it supposed to be used as the last layer, which would create this issue.
Pass the nn.ModuleList to nn.Sequential as seen here:
mlist = nn.ModuleList([
    nn.Linear(1, 2),
    nn.Linear(2, 3),
    nn.Linear(3, 4)])

model = nn.Sequential(*mlist)
model(torch.randn(1, 1))

or iterate it manually.