Advertisement
Guest User

train_val.prototxt

a guest
Jan 5th, 2015
385
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.95 KB | None | 0 0
  1. name: "LogisticRegressionNet"
  2. layers {
  3. name: "data"
  4. type: HDF5_DATA
  5. top: "data"
  6. top: "label"
  7. hdf5_data_param {
  8. source: "data/train.txt"
  9. batch_size: 1
  10. }
  11. include: { phase: TRAIN }
  12. }
  13. layers {
  14. name: "data"
  15. type: HDF5_DATA
  16. top: "data"
  17. top: "label"
  18. hdf5_data_param {
  19. source: "data/test.txt"
  20. batch_size: 1
  21. }
  22. include: { phase: TEST }
  23. }
  24. layers {
  25. name: "fc1"
  26. type: INNER_PRODUCT
  27. bottom: "data"
  28. top: "fc1"
  29. blobs_lr: 1
  30. blobs_lr: 2
  31. weight_decay: 1
  32. weight_decay: 0
  33. inner_product_param {
  34. num_output: 1
  35. weight_filler {
  36. type: "gaussian"
  37. std: 0.005
  38. }
  39. bias_filler {
  40. type: "constant"
  41. value: 0.1
  42. }
  43. }
  44. }
  45. layers {
  46. name: "loss"
  47. type: SOFTMAX_LOSS
  48. bottom: "fc1"
  49. bottom: "label"
  50. top: "loss"
  51. }
  52. layers {
  53. name: "accuracy"
  54. type: ACCURACY
  55. bottom: "fc1"
  56. bottom: "label"
  57. top: "accuracy"
  58. include: { phase: TEST }
  59. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement