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