imyhxy commited on
Commit
f64fab5
1 Parent(s): fad57c2

Fixed 'meta' and 'hyp' may out of order when using evolve (#4657)

Browse files

* Fixed 'meta' and 'hyp' may out of order when using evolve

* Update gitignore

Files changed (2) hide show
  1. .gitignore +1 -1
  2. train.py +1 -1
.gitignore CHANGED
@@ -76,7 +76,7 @@ sdist/
76
  var/
77
  wheels/
78
  *.egg-info/
79
- wandb/
80
  .installed.cfg
81
  *.egg
82
 
 
76
  var/
77
  wheels/
78
  *.egg-info/
79
+ /wandb/
80
  .installed.cfg
81
  *.egg
82
 
train.py CHANGED
@@ -570,7 +570,7 @@ def main(opt):
570
  mp, s = 0.8, 0.2 # mutation probability, sigma
571
  npr = np.random
572
  npr.seed(int(time.time()))
573
- g = np.array([x[0] for x in meta.values()]) # gains 0-1
574
  ng = len(meta)
575
  v = np.ones(ng)
576
  while all(v == 1): # mutate until a change occurs (prevent duplicates)
 
570
  mp, s = 0.8, 0.2 # mutation probability, sigma
571
  npr = np.random
572
  npr.seed(int(time.time()))
573
+ g = np.array([meta[k][0] for k in hyp.keys()]) # gains 0-1
574
  ng = len(meta)
575
  v = np.ones(ng)
576
  while all(v == 1): # mutate until a change occurs (prevent duplicates)