Advertisement
Guest User

Untitled

a guest
May 20th, 2018
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.63 KB | None | 0 0
  1. # define net structure here
  2. self.conv1 = nn.Sequential(
  3. nn.Conv2d(5, 8, (3, 3), stride=1, padding=1),
  4. nn.LeakyReLU(),
  5. nn.MaxPool2d(2)
  6. ).cuda()
  7. self.conv2 = nn.Sequential(
  8. nn.Conv2d(8, 12, (3, 3), stride=1, padding=1),
  9. nn.LeakyReLU(),
  10. nn.MaxPool2d(2)
  11. ).cuda()
  12. self.conv3 = nn.Sequential(
  13. nn.Conv2d(12, 12, (3, 3), stride=1, padding=1),
  14. nn.LeakyReLU(),
  15. nn.MaxPool2d(2)
  16. ).cuda()
  17. self.conv4 = nn.Sequential(
  18. nn.Conv2d(12, 12, (3, 3), stride=1, padding=1),
  19. nn.LeakyReLU()
  20. ).cuda()
  21. self.conv5 = nn.Sequential(
  22. nn.Conv2d(12, 12, (3, 3), stride=1, padding=1),
  23. nn.LeakyReLU()
  24. ).cuda()
  25.  
  26. n = 12*8*4
  27.  
  28. self.linear = nn.Sequential(
  29. nn.Linear(n, 100),
  30. nn.Tanh(),
  31. nn.Linear(100, 32),
  32. nn.Tanh(),
  33. nn.Linear(32, 32),
  34. nn.Tanh(),
  35. nn.Linear(32, 32),
  36. nn.Tanh(),
  37. nn.Linear(32, 32),
  38. nn.Tanh(),
  39. nn.Linear(32, 32),
  40. nn.Tanh(),
  41. nn.Linear(32, 32),
  42. nn.Tanh(),
  43. nn.Linear(32, 32),
  44. nn.Tanh(),
  45. nn.Linear(32, 32),
  46. nn.Tanh(),
  47. nn.Linear(32, 32),
  48. nn.Tanh(),
  49. nn.Linear(32, 32),
  50. nn.Tanh(),
  51. nn.Linear(32, 32),
  52. nn.Tanh(),
  53. nn.Linear(32, self.num_outputs),
  54. nn.Sigmoid(),
  55. ).cuda()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement