diff --git a/train.py b/train.py index a83fa1c0..4db2afc3 100644 --- a/train.py +++ b/train.py @@ -467,7 +467,7 @@ if __name__ == '__main__': if opt.bucket: os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket) # download evolve.txt if exists - for _ in range(100): # generations to evolve + for _ in range(1): # generations to evolve if os.path.exists('evolve.txt'): # if evolve.txt exists: select best hyps and mutate # Select parent(s) x = np.loadtxt('evolve.txt', ndmin=2) @@ -484,7 +484,7 @@ if __name__ == '__main__': # Mutate np.random.seed(int(time.time())) - s = np.random.random() * 0.15 # sigma + s = np.random.random() * 0.10 # sigma g = [1, 1, 1, 1, 1, 1, 1, 0, .1, 1, 1, 1, 1, 1, 1, 1, 1, 1] # gains for i, k in enumerate(hyp.keys()): x = (np.random.randn() * s * g[i] + 1) ** 2.0 # plt.hist(x.ravel(), 300)