optimizer_config.json 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  1. {
  2. "algorithm": "random",
  3. "parameters": {
  4. "anchor_t": {
  5. "type": "discrete",
  6. "values": [
  7. 2,
  8. 8
  9. ]
  10. },
  11. "batch_size": {
  12. "type": "discrete",
  13. "values": [
  14. 16,
  15. 32,
  16. 64
  17. ]
  18. },
  19. "box": {
  20. "type": "discrete",
  21. "values": [
  22. 0.02,
  23. 0.2
  24. ]
  25. },
  26. "cls": {
  27. "type": "discrete",
  28. "values": [
  29. 0.2
  30. ]
  31. },
  32. "cls_pw": {
  33. "type": "discrete",
  34. "values": [
  35. 0.5
  36. ]
  37. },
  38. "copy_paste": {
  39. "type": "discrete",
  40. "values": [
  41. 1
  42. ]
  43. },
  44. "degrees": {
  45. "type": "discrete",
  46. "values": [
  47. 0,
  48. 45
  49. ]
  50. },
  51. "epochs": {
  52. "type": "discrete",
  53. "values": [
  54. 5
  55. ]
  56. },
  57. "fl_gamma": {
  58. "type": "discrete",
  59. "values": [
  60. 0
  61. ]
  62. },
  63. "fliplr": {
  64. "type": "discrete",
  65. "values": [
  66. 0
  67. ]
  68. },
  69. "flipud": {
  70. "type": "discrete",
  71. "values": [
  72. 0
  73. ]
  74. },
  75. "hsv_h": {
  76. "type": "discrete",
  77. "values": [
  78. 0
  79. ]
  80. },
  81. "hsv_s": {
  82. "type": "discrete",
  83. "values": [
  84. 0
  85. ]
  86. },
  87. "hsv_v": {
  88. "type": "discrete",
  89. "values": [
  90. 0
  91. ]
  92. },
  93. "iou_t": {
  94. "type": "discrete",
  95. "values": [
  96. 0.7
  97. ]
  98. },
  99. "lr0": {
  100. "type": "discrete",
  101. "values": [
  102. 1e-05,
  103. 0.1
  104. ]
  105. },
  106. "lrf": {
  107. "type": "discrete",
  108. "values": [
  109. 0.01,
  110. 1
  111. ]
  112. },
  113. "mixup": {
  114. "type": "discrete",
  115. "values": [
  116. 1
  117. ]
  118. },
  119. "momentum": {
  120. "type": "discrete",
  121. "values": [
  122. 0.6
  123. ]
  124. },
  125. "mosaic": {
  126. "type": "discrete",
  127. "values": [
  128. 0
  129. ]
  130. },
  131. "obj": {
  132. "type": "discrete",
  133. "values": [
  134. 0.2
  135. ]
  136. },
  137. "obj_pw": {
  138. "type": "discrete",
  139. "values": [
  140. 0.5
  141. ]
  142. },
  143. "optimizer": {
  144. "type": "categorical",
  145. "values": [
  146. "SGD",
  147. "Adam",
  148. "AdamW"
  149. ]
  150. },
  151. "perspective": {
  152. "type": "discrete",
  153. "values": [
  154. 0
  155. ]
  156. },
  157. "scale": {
  158. "type": "discrete",
  159. "values": [
  160. 0
  161. ]
  162. },
  163. "shear": {
  164. "type": "discrete",
  165. "values": [
  166. 0
  167. ]
  168. },
  169. "translate": {
  170. "type": "discrete",
  171. "values": [
  172. 0
  173. ]
  174. },
  175. "warmup_bias_lr": {
  176. "type": "discrete",
  177. "values": [
  178. 0,
  179. 0.2
  180. ]
  181. },
  182. "warmup_epochs": {
  183. "type": "discrete",
  184. "values": [
  185. 5
  186. ]
  187. },
  188. "warmup_momentum": {
  189. "type": "discrete",
  190. "values": [
  191. 0,
  192. 0.95
  193. ]
  194. },
  195. "weight_decay": {
  196. "type": "discrete",
  197. "values": [
  198. 0,
  199. 0.001
  200. ]
  201. }
  202. },
  203. "spec": {
  204. "maxCombo": 0,
  205. "metric": "metrics/mAP_0.5",
  206. "objective": "maximize"
  207. },
  208. "trials": 1
  209. }