diff --git a/train.py b/train.py index ad328656..59e93619 100644 --- a/train.py +++ b/train.py @@ -43,6 +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) for k, v in zip(hyp.keys(), np.loadtxt(f[0])): hyp[k] = v