Guest User

Untitled

a guest
Mar 24th, 2018
77
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.68 KB | None | 0 0
  1. name: "mnist_deploy"
  2. op {
  3. input: "data"
  4. input: "conv1_w"
  5. input: "conv1_b"
  6. output: "conv1"
  7. name: ""
  8. type: "Conv"
  9. arg {
  10. name: "kernel"
  11. i: 5
  12. }
  13. arg {
  14. name: "order"
  15. s: "NCHW"
  16. }
  17. }
  18. op {
  19. input: "conv1"
  20. output: "pool1"
  21. name: ""
  22. type: "MaxPool"
  23. arg {
  24. name: "kernel"
  25. i: 2
  26. }
  27. arg {
  28. name: "use_cudnn"
  29. i: 0
  30. }
  31. arg {
  32. name: "stride"
  33. i: 2
  34. }
  35. }
  36. op {
  37. input: "pool1"
  38. input: "conv2_w"
  39. input: "conv2_b"
  40. output: "conv2"
  41. name: ""
  42. type: "Conv"
  43. arg {
  44. name: "kernel"
  45. i: 5
  46. }
  47. arg {
  48. name: "order"
  49. s: "NCHW"
  50. }
  51. }
  52. op {
  53. input: "conv2"
  54. output: "pool2"
  55. name: ""
  56. type: "MaxPool"
  57. arg {
  58. name: "kernel"
  59. i: 2
  60. }
  61. arg {
  62. name: "use_cudnn"
  63. i: 0
  64. }
  65. arg {
  66. name: "stride"
  67. i: 2
  68. }
  69. }
  70. op {
  71. input: "pool2"
  72. input: "fc3_w"
  73. input: "fc3_b"
  74. output: "fc3"
  75. name: ""
  76. type: "FC"
  77. arg {
  78. name: "use_cudnn"
  79. i: 0
  80. }
  81. arg {
  82. name: "order"
  83. s: "NCHW"
  84. }
  85. arg {
  86. name: "cudnn_exhaustive_search"
  87. i: 0
  88. }
  89. }
  90. op {
  91. input: "fc3"
  92. output: "relu3"
  93. name: ""
  94. type: "Relu"
  95. arg {
  96. name: "use_cudnn"
  97. i: 0
  98. }
  99. }
  100. op {
  101. input: "relu3"
  102. input: "pred_w"
  103. input: "pred_b"
  104. output: "pred"
  105. name: ""
  106. type: "FC"
  107. arg {
  108. name: "use_cudnn"
  109. i: 0
  110. }
  111. arg {
  112. name: "order"
  113. s: "NCHW"
  114. }
  115. arg {
  116. name: "cudnn_exhaustive_search"
  117. i: 0
  118. }
  119. }
  120. op {
  121. input: "pred"
  122. output: "softmax"
  123. name: ""
  124. type: "Softmax"
  125. arg {
  126. name: "use_cudnn"
  127. i: 0
  128. }
  129. }
  130. external_input: "data"
  131. external_input: "conv1_w"
  132. external_input: "conv1_b"
  133. external_input: "conv2_w"
  134. external_input: "conv2_b"
  135. external_input: "fc3_w"
  136. external_input: "fc3_b"
  137. external_input: "pred_w"
  138. external_input: "pred_b"
Add Comment
Please, Sign In to add comment