Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class ConnectNet(nn.Module):
- def __init__(self):
- super(ConnectNet, self).__init__()
- self.conv = ConvBlock()
- for block in range(19):
- setattr(self, "res_%i" % block,ResBlock())
- self.outblock = OutBlock()
- def forward(self,s):
- s = self.conv(s)
- for block in range(19):
- s = getattr(self, "res_%i" % block)(s)
- s = self.outblock(s)
- return s
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement