CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutSign UpSign In
hukaixuan19970627

Real-time collaboration for Jupyter Notebooks, Linux Terminals, LaTeX, VS Code, R IDE, and more,
all in one place. Commercial Alternative to JupyterHub.

GitHub Repository: hukaixuan19970627/yolov5_obb
 Path: blob/master/utils/loggers/wandb/sweep.yaml
Views: 475
1
# Hyperparameters for training

2
# To set range-

3
# Provide min and max values as:

4
#      parameter:

5
#

6
#         min: scalar

7
#         max: scalar

8
# OR

9
#

10
# Set a specific list of search space-

11
#     parameter:

12
#         values: [scalar1, scalar2, scalar3...]

13
#

14
# You can use grid, bayesian and hyperopt search strategy

15
# For more info on configuring sweeps visit - https://docs.wandb.ai/guides/sweeps/configuration

16


17
program: utils/loggers/wandb/sweep.py

18
method: random

19
metric:

20
  name: metrics/mAP_0.5

21
  goal: maximize

22


23
parameters:

24
  # hyperparameters: set either min, max range or values list

25
  data:

26
    value: "data/coco128.yaml"

27
  batch_size:

28
    values: [64]

29
  epochs:

30
    values: [10]

31


32
  lr0:

33
    distribution: uniform

34
    min: 1e-5

35
    max: 1e-1

36
  lrf:

37
    distribution: uniform

38
    min: 0.01

39
    max: 1.0

40
  momentum:

41
    distribution: uniform

42
    min: 0.6

43
    max: 0.98

44
  weight_decay:

45
    distribution: uniform

46
    min: 0.0

47
    max: 0.001

48
  warmup_epochs:

49
    distribution: uniform

50
    min: 0.0

51
    max: 5.0

52
  warmup_momentum:

53
    distribution: uniform

54
    min: 0.0

55
    max: 0.95

56
  warmup_bias_lr:

57
    distribution: uniform

58
    min: 0.0

59
    max: 0.2

60
  box:

61
    distribution: uniform

62
    min: 0.02

63
    max: 0.2

64
  cls:

65
    distribution: uniform

66
    min: 0.2

67
    max: 4.0

68
  cls_pw:

69
    distribution: uniform

70
    min: 0.5

71
    max: 2.0

72
  obj:

73
    distribution: uniform

74
    min: 0.2

75
    max: 4.0

76
  obj_pw:

77
    distribution: uniform

78
    min: 0.5

79
    max: 2.0

80
  iou_t:

81
    distribution: uniform

82
    min: 0.1

83
    max: 0.7

84
  anchor_t:

85
    distribution: uniform

86
    min: 2.0

87
    max: 8.0

88
  fl_gamma:

89
    distribution: uniform

90
    min: 0.0

91
    max: 0.1

92
  hsv_h:

93
    distribution: uniform

94
    min: 0.0

95
    max: 0.1

96
  hsv_s:

97
    distribution: uniform

98
    min: 0.0

99
    max: 0.9

100
  hsv_v:

101
    distribution: uniform

102
    min: 0.0

103
    max: 0.9

104
  degrees:

105
    distribution: uniform

106
    min: 0.0

107
    max: 45.0

108
  translate:

109
    distribution: uniform

110
    min: 0.0

111
    max: 0.9

112
  scale:

113
    distribution: uniform

114
    min: 0.0

115
    max: 0.9

116
  shear:

117
    distribution: uniform

118
    min: 0.0

119
    max: 10.0

120
  perspective:

121
    distribution: uniform

122
    min: 0.0

123
    max: 0.001

124
  flipud:

125
    distribution: uniform

126
    min: 0.0

127
    max: 1.0

128
  fliplr:

129
    distribution: uniform

130
    min: 0.0

131
    max: 1.0

132
  mosaic:

133
    distribution: uniform

134
    min: 0.0

135
    max: 1.0

136
  mixup:

137
    distribution: uniform

138
    min: 0.0

139
    max: 1.0

140
  copy_paste:

141
    distribution: uniform

142
    min: 0.0

143
    max: 1.0

144


145