Guest User

Untitled

a guest
Nov 24th, 2017
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.19 KB | None | 0 0
  1. name: "iris"
  2. layer {
  3. name: "data"
  4. type: "HDF5Data"
  5. top: "data"
  6. top: "label"
  7. include: {
  8. phase: TRAIN
  9. }
  10. hdf5_data_param {
  11. source: "./files/train_file_location.txt"
  12. batch_size: 10
  13. }
  14. }
  15. layer {
  16. name: "data"
  17. type: "HDF5Data"
  18. top: "data"
  19. top: "label"
  20. include: {
  21. phase: TEST
  22. }
  23. hdf5_data_param {
  24. source: "./files/test_file_location.txt"
  25. batch_size: 10
  26. }
  27. }
  28. layer {
  29. name: "ip1"
  30. type: "InnerProduct"
  31. bottom: "data"
  32. top: "ip1"
  33. inner_product_param {
  34. num_output: 20
  35. weight_filler {
  36. type: "xavier"
  37. }
  38. bias_filler {
  39. type: "constant"
  40. }
  41. }
  42. }
  43. layer {
  44. name: "relu1"
  45. type: "ReLU"
  46. bottom: "ip1"
  47. top: "ip1"
  48. }
  49. layer {
  50. name: "ip2"
  51. type: "InnerProduct"
  52. bottom: "ip1"
  53. top: "ip2"
  54. inner_product_param {
  55. num_output: 3
  56. weight_filler {
  57. type: "xavier"
  58. }
  59. bias_filler {
  60. type: "constant"
  61. }
  62. }
  63. }
  64. layer {
  65. name: "softmax"
  66. type: "Softmax"
  67. bottom: "ip2"
  68. top: "smip2"
  69. }
  70.  
  71. layer {
  72. name: "loss"
  73. type: "SoftmaxWithLoss"
  74. bottom: "ip2"
  75. bottom: "label"
  76. top: "loss"
  77. }
  78.  
  79. layer {
  80. name: "accuracy"
  81. type: "Accuracy"
  82. bottom: "smip2"
  83. bottom: "label"
  84. top: "accuracy"
  85. include {
  86. phase: TEST
  87. }
  88. }
Add Comment
Please, Sign In to add comment