updates
This commit is contained in:
parent
1cfd370455
commit
8ae16adde4
2
train.py
2
train.py
|
@ -393,7 +393,7 @@ if __name__ == '__main__':
|
||||||
|
|
||||||
# Clip to limits
|
# Clip to limits
|
||||||
keys = ['lr0', 'iou_t', 'momentum', 'weight_decay', 'hsv_s', 'hsv_v', 'translate', 'scale']
|
keys = ['lr0', 'iou_t', 'momentum', 'weight_decay', 'hsv_s', 'hsv_v', 'translate', 'scale']
|
||||||
limits = [(1e-4, 1e-2), (0.00, 0.70), (0.60, 0.95), (0, 0.001), (0, .8), (0, .8), (0, .8), (0, .8)]
|
limits = [(1e-4, 1e-2), (0.00, 0.70), (0.60, 0.99), (0, 0.001), (0, .8), (0, .8), (0, .8), (0, .8)]
|
||||||
for k, v in zip(keys, limits):
|
for k, v in zip(keys, limits):
|
||||||
hyp[k] = np.clip(hyp[k], v[0], v[1])
|
hyp[k] = np.clip(hyp[k], v[0], v[1])
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue