From c029f8eabf8aa3d675bcccd2778e5467bf161c61 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 5 Jul 2021 12:44:20 +0200 Subject: [PATCH] Evolution commented `hyp['anchors']` fix Fix for #3886 --- train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train.py b/train.py index 2e864a60cefc..5a434773eff7 100644 --- a/train.py +++ b/train.py @@ -596,6 +596,8 @@ def main(opt): with open(opt.hyp) as f: hyp = yaml.safe_load(f) # load hyps dict + if 'anchors' not in hyp: # anchors commented in hyp.yaml + hyp['anchors'] = 3 assert LOCAL_RANK == -1, 'DDP mode not implemented for --evolve' opt.notest, opt.nosave = True, True # only test/save final epoch # ei = [isinstance(x, (int, float)) for x in hyp.values()] # evolvable indices