Advertisement
Guest User

Untitled

a guest
Apr 25th, 2019
75
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.42 KB | None | 0 0
  1. class ConnectNet(nn.Module):
  2. def __init__(self):
  3. super(ConnectNet, self).__init__()
  4. self.conv = ConvBlock()
  5. for block in range(19):
  6. setattr(self, "res_%i" % block,ResBlock())
  7. self.outblock = OutBlock()
  8.  
  9. def forward(self,s):
  10. s = self.conv(s)
  11. for block in range(19):
  12. s = getattr(self, "res_%i" % block)(s)
  13. s = self.outblock(s)
  14. return s
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement