alexnet.cfg 879 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. [net]
  2. batch=128
  3. subdivisions=1
  4. height=227
  5. width=227
  6. channels=3
  7. momentum=0.9
  8. decay=0.0005
  9. max_crop=256
  10. learning_rate=0.01
  11. policy=poly
  12. power=4
  13. max_batches=800000
  14. angle=7
  15. hue = .1
  16. saturation=.75
  17. exposure=.75
  18. aspect=.75
  19. [convolutional]
  20. filters=96
  21. size=11
  22. stride=4
  23. pad=0
  24. activation=relu
  25. [maxpool]
  26. size=3
  27. stride=2
  28. padding=0
  29. [convolutional]
  30. filters=256
  31. size=5
  32. stride=1
  33. pad=1
  34. activation=relu
  35. [maxpool]
  36. size=3
  37. stride=2
  38. padding=0
  39. [convolutional]
  40. filters=384
  41. size=3
  42. stride=1
  43. pad=1
  44. activation=relu
  45. [convolutional]
  46. filters=384
  47. size=3
  48. stride=1
  49. pad=1
  50. activation=relu
  51. [convolutional]
  52. filters=256
  53. size=3
  54. stride=1
  55. pad=1
  56. activation=relu
  57. [maxpool]
  58. size=3
  59. stride=2
  60. padding=0
  61. [connected]
  62. output=4096
  63. activation=relu
  64. [dropout]
  65. probability=.5
  66. [connected]
  67. output=4096
  68. activation=relu
  69. [dropout]
  70. probability=.5
  71. [connected]
  72. output=1000
  73. activation=linear
  74. [softmax]
  75. groups=1
  76. [cost]
  77. type=sse