updates
This commit is contained in:
parent
84c1fecd81
commit
1096596ad8
2
train.py
2
train.py
|
@ -367,7 +367,7 @@ if __name__ == '__main__':
|
||||||
|
|
||||||
# Clip to limits
|
# Clip to limits
|
||||||
keys = ['lr0', 'iou_t', 'momentum', 'weight_decay']
|
keys = ['lr0', 'iou_t', 'momentum', 'weight_decay']
|
||||||
limits = [(1e-4, 1e-2), (0, 0.90), (0.70, 0.99), (0, 0.01)]
|
limits = [(1e-4, 1e-2), (0, 0.00), (0.70, 0.99), (0, 0.01)]
|
||||||
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