ptrblck:

nn.Module you should use nn.ModuleList


You mean lins = [nn.Linear(4, 1).cuda() for _ in range(8)] to
self.lins = nn.ModuleList([])
for i in range (8): 
   lins.append(nn.Linear(4, 1))

Am I correct?