ashwinnair14

Untitled

Dec 14th, 2015
193
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 5.53 KB | None | 0 0
  1. name: "AlexNet_mod"
  2. layer {
  3. name: "data"
  4. type: "HDF5Data"
  5. top: "data"
  6. top: "label"
  7. top: "data2"
  8. include {
  9. phase: TRAIN
  10. }
  11. hdf5_data_param {
  12. source: "/home/anilil/caffe/models/multistream/train.txt"
  13. batch_size: 10
  14. }
  15. }
  16. layer {
  17. name: "data"
  18. type: "HDF5Data"
  19. top: "data"
  20. top: "data2"
  21. top: "label"
  22. include {
  23. phase: TEST
  24. }
  25. hdf5_data_param {
  26. source: "/home/anilil/caffe/models/multistream/test.txt"
  27. batch_size: 5
  28. }
  29. }
  30. layer {
  31. name: "conv1"
  32. type: "Convolution"
  33. bottom: "data"
  34. top: "conv1"
  35. param {
  36. lr_mult: 1
  37. decay_mult: 1
  38. }
  39. param {
  40. lr_mult: 2
  41. decay_mult: 0
  42. }
  43. convolution_param {
  44. num_output: 96
  45. kernel_size: 11
  46. stride: 4
  47. weight_filler {
  48. type: "gaussian"
  49. std: 0.01
  50. }
  51. bias_filler {
  52. type: "constant"
  53. value: 0
  54. }
  55. }
  56. }
  57. layer {
  58. name: "relu1"
  59. type: "ReLU"
  60. bottom: "conv1"
  61. top: "conv1"
  62. }
  63. layer {
  64. name: "norm1"
  65. type: "LRN"
  66. bottom: "conv1"
  67. top: "norm1"
  68. lrn_param {
  69. local_size: 5
  70. alpha: 0.0001
  71. beta: 0.75
  72. }
  73. }
  74. layer {
  75. name: "pool1"
  76. type: "Pooling"
  77. bottom: "norm1"
  78. top: "pool1"
  79. pooling_param {
  80. pool: MAX
  81. kernel_size: 3
  82. stride: 2
  83. }
  84. }
  85. layer {
  86. name: "conv2"
  87. type: "Convolution"
  88. bottom: "pool1"
  89. top: "conv2"
  90. param {
  91. lr_mult: 1
  92. decay_mult: 1
  93. }
  94. param {
  95. lr_mult: 2
  96. decay_mult: 0
  97. }
  98. convolution_param {
  99. num_output: 256
  100. pad: 2
  101. kernel_size: 5
  102. group: 2
  103. weight_filler {
  104. type: "gaussian"
  105. std: 0.01
  106. }
  107. bias_filler {
  108. type: "constant"
  109. value: 0.1
  110. }
  111. }
  112. }
  113. layer {
  114. name: "relu2"
  115. type: "ReLU"
  116. bottom: "conv2"
  117. top: "conv2"
  118. }
  119. layer {
  120. name: "norm2"
  121. type: "LRN"
  122. bottom: "conv2"
  123. top: "norm2"
  124. lrn_param {
  125. local_size: 5
  126. alpha: 0.0001
  127. beta: 0.75
  128. }
  129. }
  130. layer {
  131. name: "pool2"
  132. type: "Pooling"
  133. bottom: "norm2"
  134. top: "pool2"
  135. pooling_param {
  136. pool: MAX
  137. kernel_size: 3
  138. stride: 2
  139. }
  140. }
  141. layer {
  142. name: "conv3"
  143. type: "Convolution"
  144. bottom: "pool2"
  145. top: "conv3"
  146. param {
  147. lr_mult: 1
  148. decay_mult: 1
  149. }
  150. param {
  151. lr_mult: 2
  152. decay_mult: 0
  153. }
  154. convolution_param {
  155. num_output: 384
  156. pad: 1
  157. kernel_size: 3
  158. weight_filler {
  159. type: "gaussian"
  160. std: 0.01
  161. }
  162. bias_filler {
  163. type: "constant"
  164. value: 0
  165. }
  166. }
  167. }
  168. layer {
  169. name: "relu3"
  170. type: "ReLU"
  171. bottom: "conv3"
  172. top: "conv3"
  173. }
  174. layer {
  175. name: "conv4"
  176. type: "Convolution"
  177. bottom: "conv3"
  178. top: "conv4"
  179. param {
  180. lr_mult: 1
  181. decay_mult: 1
  182. }
  183. param {
  184. lr_mult: 2
  185. decay_mult: 0
  186. }
  187. convolution_param {
  188. num_output: 384
  189. pad: 1
  190. kernel_size: 3
  191. group: 2
  192. weight_filler {
  193. type: "gaussian"
  194. std: 0.01
  195. }
  196. bias_filler {
  197. type: "constant"
  198. value: 0.1
  199. }
  200. }
  201. }
  202. layer {
  203. name: "relu4"
  204. type: "ReLU"
  205. bottom: "conv4"
  206. top: "conv4"
  207. }
  208. layer {
  209. name: "conv5"
  210. type: "Convolution"
  211. bottom: "conv4"
  212. top: "conv5"
  213. param {
  214. lr_mult: 1
  215. decay_mult: 1
  216. }
  217. param {
  218. lr_mult: 2
  219. decay_mult: 0
  220. }
  221. convolution_param {
  222. num_output: 256
  223. pad: 1
  224. kernel_size: 3
  225. group: 2
  226. weight_filler {
  227. type: "gaussian"
  228. std: 0.01
  229. }
  230. bias_filler {
  231. type: "constant"
  232. value: 0.1
  233. }
  234. }
  235. }
  236. layer {
  237. name: "relu5"
  238. type: "ReLU"
  239. bottom: "conv5"
  240. top: "conv5"
  241. }
  242. layer {
  243. name: "pool5"
  244. type: "Pooling"
  245. bottom: "conv5"
  246. top: "pool5"
  247. pooling_param {
  248. pool: MAX
  249. kernel_size: 3
  250. stride: 2
  251. }
  252. }
  253. layer {
  254. name: "fc6"
  255. type: "InnerProduct"
  256. bottom: "pool5"
  257. top: "fc6"
  258. param {
  259. lr_mult: 1
  260. decay_mult: 1
  261. }
  262. param {
  263. lr_mult: 2
  264. decay_mult: 0
  265. }
  266. inner_product_param {
  267. num_output: 4096
  268. weight_filler {
  269. type: "gaussian"
  270. std: 0.005
  271. }
  272. bias_filler {
  273. type: "constant"
  274. value: 0.1
  275. }
  276. }
  277. }
  278. layer {
  279. name: "relu6"
  280. type: "ReLU"
  281. bottom: "fc6"
  282. top: "fc6"
  283. }
  284. layer {
  285. name: "drop6"
  286. type: "Dropout"
  287. bottom: "fc6"
  288. top: "fc6"
  289. dropout_param {
  290. dropout_ratio: 0.5
  291. }
  292. }
  293. layer {
  294. name: "concat"
  295. bottom: "fc6"
  296. bottom: "data2"
  297. top: "out"
  298. type: "Concat"
  299. concat_param {
  300. axis: 1
  301. }
  302. }
  303. layer {
  304. name: "fc7"
  305. type: "InnerProduct"
  306. bottom: "out"
  307. top: "fc7"
  308. param {
  309. lr_mult: 1
  310. decay_mult: 1
  311. }
  312. param {
  313. lr_mult: 2
  314. decay_mult: 0
  315. }
  316. inner_product_param {
  317. num_output: 4096
  318. weight_filler {
  319. type: "gaussian"
  320. std: 0.005
  321. }
  322. bias_filler {
  323. type: "constant"
  324. value: 0.1
  325. }
  326. }
  327. }
  328. layer {
  329. name: "relu7"
  330. type: "ReLU"
  331. bottom: "fc7"
  332. top: "fc7"
  333. }
  334. layer {
  335. name: "drop7"
  336. type: "Dropout"
  337. bottom: "fc7"
  338. top: "fc7"
  339. dropout_param {
  340. dropout_ratio: 0.5
  341. }
  342. }
  343. layer {
  344. name: "fc8"
  345. type: "InnerProduct"
  346. bottom: "fc7"
  347. top: "fc8"
  348. param {
  349. lr_mult: 1
  350. decay_mult: 1
  351. }
  352. param {
  353. lr_mult: 2
  354. decay_mult: 0
  355. }
  356. inner_product_param {
  357. num_output: 1000
  358. weight_filler {
  359. type: "gaussian"
  360. std: 0.01
  361. }
  362. bias_filler {
  363. type: "constant"
  364. value: 0
  365. }
  366. }
  367. }
  368. layer {
  369. name: "accuracy"
  370. type: "Accuracy"
  371. bottom: "fc8"
  372. bottom: "label"
  373. top: "accuracy"
  374. include {
  375. phase: TEST
  376. }
  377. }
  378. layer {
  379. name: "loss"
  380. type: "SoftmaxWithLoss"
  381. bottom: "fc8"
  382. bottom: "label"
  383. top: "loss"
  384. }
Advertisement
Add Comment
Please, Sign In to add comment