This commit is contained in:
Glenn Jocher 2019-07-22 16:14:41 +02:00
parent 1cfd370455
commit 8ae16adde4
1 changed files with 1 additions and 1 deletions

View File

@ -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])