diff --git a/train.py b/train.py index 59e93619..67996e4e 100644 --- a/train.py +++ b/train.py @@ -43,7 +43,7 @@ hyp = {'giou': 3.31, # giou loss gain # Overwrite hyp with hyp*.txt (optional) f = glob.glob('hyp*.txt') if f: - print('Using %s' % f) + print('Using %s' % f[0]) for k, v in zip(hyp.keys(), np.loadtxt(f[0])): hyp[k] = v