cifar.cfg 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. [net]
  2. batch=128
  3. subdivisions=1
  4. height=28
  5. width=28
  6. channels=3
  7. max_crop=32
  8. min_crop=32
  9. hue=.1
  10. saturation=.75
  11. exposure=.75
  12. learning_rate=0.4
  13. policy=poly
  14. power=4
  15. max_batches = 5000
  16. momentum=0.9
  17. decay=0.0005
  18. [convolutional]
  19. batch_normalize=1
  20. filters=128
  21. size=3
  22. stride=1
  23. pad=1
  24. activation=leaky
  25. [convolutional]
  26. batch_normalize=1
  27. filters=128
  28. size=3
  29. stride=1
  30. pad=1
  31. activation=leaky
  32. [convolutional]
  33. batch_normalize=1
  34. filters=128
  35. size=3
  36. stride=1
  37. pad=1
  38. activation=leaky
  39. [maxpool]
  40. size=2
  41. stride=2
  42. [dropout]
  43. probability=.5
  44. [convolutional]
  45. batch_normalize=1
  46. filters=256
  47. size=3
  48. stride=1
  49. pad=1
  50. activation=leaky
  51. [convolutional]
  52. batch_normalize=1
  53. filters=256
  54. size=3
  55. stride=1
  56. pad=1
  57. activation=leaky
  58. [convolutional]
  59. batch_normalize=1
  60. filters=256
  61. size=3
  62. stride=1
  63. pad=1
  64. activation=leaky
  65. [maxpool]
  66. size=2
  67. stride=2
  68. [dropout]
  69. probability=.5
  70. [convolutional]
  71. batch_normalize=1
  72. filters=512
  73. size=3
  74. stride=1
  75. pad=1
  76. activation=leaky
  77. [convolutional]
  78. batch_normalize=1
  79. filters=512
  80. size=3
  81. stride=1
  82. pad=1
  83. activation=leaky
  84. [convolutional]
  85. batch_normalize=1
  86. filters=512
  87. size=3
  88. stride=1
  89. pad=1
  90. activation=leaky
  91. [dropout]
  92. probability=.5
  93. [convolutional]
  94. filters=10
  95. size=1
  96. stride=1
  97. pad=1
  98. activation=leaky
  99. [avgpool]
  100. [softmax]
  101. groups=1