def __init__(self, n):
"""Even n preferred"""
super(flowGAN, self).__init__()
self.lin = nn.ModuleList([LU(2) for i in range(n + 1)])
hyp = []
for i in range(n):
if i % 2 == 0:
hyp.append(nLayer(2))
else:
hyp.append(nLayer(2))
# hyp.append(backLayer(2))
self.hyp = nn.ModuleList(hyp)
self.n = n
def __init__(self):
super(layer1, self).__init__()
self.fc1 = LU(2)
self.nln = nLayer(2)
self.fc2 = LU(2)