Fixed 'meta' and 'hyp' may out of order when using evolve (ultralytic…

…s#4657) * Fixed 'meta' and 'hyp' may out of order when using evolve * Update gitignore
fhkiel-mlaip · Sep 5, 2021 · e80e3d7 · e80e3d7
1 parent 06521cf
commit e80e3d7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -76,7 +76,7 @@ sdist/
 var/
 wheels/
 *.egg-info/
-wandb/
+/wandb/
 .installed.cfg
 *.egg
 

diff --git a/train.py b/train.py
@@ -570,7 +570,7 @@ def main(opt):
                 mp, s = 0.8, 0.2  # mutation probability, sigma
                 npr = np.random
                 npr.seed(int(time.time()))
-                g = np.array([x[0] for x in meta.values()])  # gains 0-1
+                g = np.array([meta[k][0] for k in hyp.keys()])  # gains 0-1
                 ng = len(meta)
                 v = np.ones(ng)
                 while all(v == 1):  # mutate until a change occurs (prevent duplicates)